Preparing submission

requirements.txt

@@ -14,4 +14,4 @@ torch==2.5.1
 torchvision==0.20.1
 onnx==1.17.0
 onnxslim==0.1.48
-onnxruntime==1.20.1
+onnxruntime

tasks/image.py

@@ -112,7 +112,7 @@ async def evaluate_image(request: ImageEvaluationRequest):
     # print("CUDA available, loading FP16 pytorch model")
     model_name = "pruned_fp16.pt"
     model = YOLO(Path(model_path, model_name), task="detect")
-    # device = device("cuda")
+    # device_name = device("cuda")
     device_name = device("cuda" if is_available() else "cpu")
     IMGSIZE = 1280
     # # If not, load FP16 ONNX model
@@ -120,7 +120,7 @@ async def evaluate_image(request: ImageEvaluationRequest):
     #     print("CUDA not, available, loading ONNX model")
     #     model_name = "640_fp16_cpu.onnx"
     #     model = YOLO(Path(model_path, model_name), task="detect")
-    #     device = device("cpu")
+    #     device_name = device("cpu")
     #     IMGSIZE = 640 # required to make CPU inference a bit fast 
 
     predictions = []