Update app.py
Browse files
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
|