1 commit

app.py

@@ -3,6 +3,7 @@ import onemetric
 import supervision
 import typing
 import tqdm
+import os
 from ultralytics import YOLO
 from dataclasses import dataclass
 from onemetric.cv.utils.iou import box_iou_batch
@@ -13,6 +14,8 @@ from supervision import Color
 from supervision import VideoInfo
 from supervision import get_video_frames_generator
 from supervision import VideoSink
+os.system("pip -qq install yolox")
+import torch
 from typing import List
 import numpy as np
 import gradio as gr

requirements.txt

@@ -3,7 +3,6 @@
 
 # Base ----------------------------------------
 torch
-yolox==0.2.0
 matplotlib>=3.2.2
 opencv-python>=4.6.0
 pillow>=7.1.2