Update app.py

app.py

@@ -5,7 +5,7 @@ from detect_strongsort import run_strongsort
 from detect import run
 import os
 import threading
-
+import torch
 should_continue = True
 
 
@@ -34,7 +34,8 @@ def yolov9_inference(model_id, img_path=None, vid_path=None, tracking_algorithm
         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='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)
+            device_strongsort = torch.device('cuda:0')
+            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=device_strongsort, 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='0', hide_conf= True)
         # Assuming output_path is the path to the output file