kadirnar commited on
Commit
8458b85
·
verified ·
1 Parent(s): 55311cc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ def yolov9_inference(img_path, model_path,image_size, conf_threshold, iou_thresh
18
  import yolov9
19
 
20
  # Load the model
21
- model = yolov9.load(model_path, device="cpu")
22
 
23
  # Set model parameters
24
  model.conf = conf_threshold
 
18
  import yolov9
19
 
20
  # Load the model
21
+ model = yolov9.load(model_path, device="cuda:0")
22
 
23
  # Set model parameters
24
  model.conf = conf_threshold