Update app.py

app.py

@@ -18,7 +18,7 @@ def yolov9_inference(img_path, conf_threshold, iou_threshold):
     
     # Load the model
     model_path = "yolov9-plant.pt" # model
-    model = yolov9.load(model_path, device="cuda")
+    model = yolov9.load(model_path, device="cuda:0")
     
     # Set model parameters
     model.conf = conf_threshold