Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -17,7 +17,7 @@ def yolov9_inference(img_path, model_id, image_size, conf_threshold, iou_thresho
|
|
17 |
|
18 |
# Load the model
|
19 |
model_path = download_models(model_id)
|
20 |
-
model = yolov9.load(model_path, device="
|
21 |
|
22 |
# Set model parameters
|
23 |
model.conf = conf_threshold
|
|
|
17 |
|
18 |
# Load the model
|
19 |
model_path = download_models(model_id)
|
20 |
+
model = yolov9.load(model_path, device="cpu")
|
21 |
|
22 |
# Set model parameters
|
23 |
model.conf = conf_threshold
|