@@ -15,7 +15,7 @@ YOLO_DIR = '/workspace/yolov5'
WEIGHTS = '/workspace/best.pt'
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-print(device)
+print('====',torch.cuda.is_available())
app = FastAPI()
templates = Jinja2Templates(directory = 'templates')
