sfoy commited on
Commit
bea8391
·
verified ·
1 Parent(s): 16e71ba

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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="cuda:0")
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