Retrun to 1 GPU inference

app.py

@@ -22,7 +22,7 @@ def yolov9_inference(model_id, image_size, conf_threshold, iou_threshold, img_pa
         if vid_extension.lower() in vid_extensions:
             input_path = vid_path
 
-    output_path = run(yolo_weights=model_id, imgsz=(image_size,image_size), conf_thres=conf_threshold, iou_thres=iou_threshold, source=input_path, device='0, 1', strong_sort_weights = "osnet_x0_25_msmt17.pt", hide_conf= True)
+    output_path = run(yolo_weights=model_id, imgsz=(image_size,image_size), conf_thres=conf_threshold, iou_thres=iou_threshold, source=input_path, device='0', strong_sort_weights = "osnet_x0_25_msmt17.pt", hide_conf= True)
     # Assuming output_path is the path to the output file
     _, output_extension = os.path.splitext(output_path)
     if output_extension.lower() in img_extensions: