|
|
@@ -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')
|
