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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -13
app.py CHANGED
@@ -1,20 +1,11 @@
 
 
 
1
  import gradio as gr
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
 
1
+ import matplotlib
2
+ matplotlib.use('Agg') # Use the 'Agg' backend for headless environments
3
+
4
  import gradio as gr
5
  import cv2
6
  import numpy as np
7
  from ultralytics import YOLO
8
+ from sort import Sort # SORT tracker
 
 
 
 
 
 
 
 
 
 
 
 
9
 
10
 
11
  #from sort import Sort # SORT tracker