使用cuda版torch进行训练和推理

app/application/app.py

@@ -1,6 +1,7 @@
 from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
 import asyncio
+import torch
 
 from app.application.token_middleware import TokenMiddleware
 from app.application.logger_middleware import LoggerMiddleware