1 commit

app.py

@@ -24,16 +24,29 @@ import yolox
 os.system("pip3 install cython_bbox gdown 'git+https://github.com/cocodataset/cocoapi.git#subdirectory=PythonAPI'")
 os.system("pip3 install -v -e .")
 from yolox.tracker.byte_tracker import BYTETracker, STrack
-
+import requests
 
 MODEL = "./best.pt"
 
-SOURCE_VIDEO_PATH = "./examples"
-
 TARGET_VIDEO_PATH = "test.mp4"
 
 CLASS_ID = [0,1,2,3]
 
+file_urls = [
+    'https://www.dropbox.com/s/qf9le18eoirfkbd/3_2.mp4?dl=0'
+]
+
+def download_file(url, save_name):
+    url = url
+    if not os.path.exists(save_name):
+        file = requests.get(url)
+        open(save_name, 'wb').write(file.content)
+
+for i, url in enumerate(file_urls):
+    download_file(file_urls[i], f"video.mp4")
+
+video_path = [['video.mp4']]
+
 model = YOLO(MODEL)
 model.fuse()
 
@@ -128,5 +141,5 @@ def ObjectDetection(video_path):
 
     return TARGET_VIDEO_PATH
 
-demo = gr.Interface(fn=ObjectDetection, inputs=gr.Video(), outputs=gr.Video(), examples=SOURCE_VIDEO_PATH, cache_examples=False)
+demo = gr.Interface(fn=ObjectDetection, inputs=gr.Video(), outputs=gr.Video(), examples=video_path, cache_examples=False)
 demo.launch()

requirements.txt

@@ -82,4 +82,5 @@ lap
 motmetrics
 filterpy
 h5py
-cython
+cython
+requests