1 commit

app.py

@@ -15,13 +15,13 @@ from supervision import VideoInfo
 from supervision import get_video_frames_generator
 from supervision import VideoSink
 import torch
-os.system("pip install git+https://github.com/Megvii-BaseDetection/YOLOX")
+os.system("pip install git+https://github.com/ifzhang/ByteTrack")
 from typing import List
 import numpy as np
 import gradio as gr
 from tqdm import tqdm
 import yolox
-from yolox import BYTETracker, STrack
+from yolox.tracker.byte_tracker import BYTETracker, STrack
 
 
 MODEL = "./best.pt"