hb-setosys commited on
Commit
aebce39
·
verified ·
1 Parent(s): ad605e9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +16 -1
app.py CHANGED
@@ -2,7 +2,22 @@ import gradio as gr
2
  import cv2
3
  import numpy as np
4
  from ultralytics import YOLO
5
- from sort import Sort # SORT tracker
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
6
 
7
  # Load YOLOv8 model (pre-trained on COCO dataset)
8
  model = YOLO("yolov8x.pt") # Highest accuracy version
 
2
  import cv2
3
  import numpy as np
4
  from ultralytics import YOLO
5
+
6
+ import os
7
+ import subprocess
8
+
9
+ # Clone the SORT repository
10
+ if not os.path.isdir("sort"):
11
+ subprocess.run(["git", "clone", "https://github.com/abewley/sort.git"])
12
+
13
+ # Add sort module to the Python path
14
+ import sys
15
+ sys.path.append(os.path.join(os.getcwd(), "sort"))
16
+
17
+ from sort import Sort
18
+
19
+
20
+ #from sort import Sort # SORT tracker
21
 
22
  # Load YOLOv8 model (pre-trained on COCO dataset)
23
  model = YOLO("yolov8x.pt") # Highest accuracy version