fix error

app.py

@@ -25,18 +25,18 @@ def yolov9_inference(model_id, img_path=None, vid_path=None, tracking_algorithm
         #if img_extension.lower() in img_extensions:
         input_path = img_path
         print(input_path)
-        output_path = run(weights=model_id, imgsz=(image_size,image_size), conf_thres=conf_threshold, iou_thres=iou_threshold, source=input_path, device='cpu', hide_conf= True)
+        output_path = run(weights=model_id, imgsz=(image_size,image_size), conf_thres=conf_threshold, iou_thres=iou_threshold, source=input_path, device='0', hide_conf= True)
     elif vid_path is not None:
         #_, vid_extension = os.path.splitext(vid_path)
         #if vid_extension.lower() in vid_extensions:
         input_path = vid_path
         print(input_path)
         if tracking_algorithm == 'deep_sort':
-            output_path = run_deepsort(weights=model_id, imgsz=(image_size,image_size), conf_thres=conf_threshold, iou_thres=iou_threshold, source=input_path, device='cpu', draw_trails=True)
+            output_path = run_deepsort(weights=model_id, imgsz=(image_size,image_size), conf_thres=conf_threshold, iou_thres=iou_threshold, source=input_path, device='0', draw_trails=True)
         elif tracking_algorithm == 'strong_sort':
             output_path = run_strongsort(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)
         else: 
-            output_path =  run(weights=model_id, imgsz=(image_size,image_size), conf_thres=conf_threshold, iou_thres=iou_threshold, source=input_path, device='cpu', hide_conf= True)
+            output_path =  run(weights=model_id, imgsz=(image_size,image_size), conf_thres=conf_threshold, iou_thres=iou_threshold, source=input_path, device='0', 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:

requirements.txt

@@ -11,7 +11,6 @@ Pillow>=7.1.2
 psutil
 torchreid
 gdown
-=======
 PyYAML>=5.3.1
 requests>=2.23.0
 scipy>=1.4.1