Update app.py

app.py

@@ -36,7 +36,7 @@ def yolo_inference(image, model_id, conf_threshold, iou_threshold, max_detection
     # Download models
     model_path = download_models(repo_id, model_id)
     
-    model = YOLO(model_path, device = "cuda:0" if torch.cuda.is_available() else "cpu")
+    model = YOLO(model_path)
     results = model(source=image, imgsz=640, iou=iou_threshold, conf=conf_threshold, verbose=False, max_det=max_detection)[0]
     detections = sv.Detections.from_ultralytics(results)