update

app.py

@@ -14,7 +14,6 @@ from huggingface_hub import HfApi
     
 def yolov7_inference(
     image: gr.inputs.Image = None,
-    video: gr.inputs.Video = None,
     model_path: gr.inputs.Dropdown = None,
     image_size: gr.inputs.Slider = 640,
     conf_threshold: gr.inputs.Slider = 0.25,
@@ -24,7 +23,6 @@ def yolov7_inference(
     YOLOv7 inference function
     Args:
         image: Input image
-        video: Input video
         model_path: Path to the model
         image_size: Image size
         conf_threshold: Confidence threshold
@@ -36,9 +34,28 @@ def yolov7_inference(
     model = yolov7.load(model_path, device="cpu", hf_model=True, trace=False)
     model.conf = conf_threshold
     model.iou = iou_threshold
-    results = model([image], [video], size=image_size)
+    results = model([image], size=image_size)
     return results.render()[0]
         
+def yolov7_inference_video(
+    video: gr.inputs.Video = None,
+    model_path: gr.inputs.Dropdown = None,
+    
+):
+    """
+    YOLOv7 inference function
+    Args:
+        video: Input video
+        model_path: Path to the model
+    
+    Returns:
+        Rendered video
+    """
+
+    model = yolov7.load(model_path, device="cpu", hf_model=True, trace=False)
+    results = model([video])
+    return results
+
 
 inputs = [
     gr.inputs.Image(type="pil", label="Input Image"),
@@ -70,4 +87,14 @@ demo_app = gr.Interface(
     cache_examples=True,
     theme='huggingface',
 )
+demo_app_video = gr.Interface(
+    fn=yolov7_inference_video,
+    inputs=inputs,
+    outputs=outputs,
+    title=title,
+    examples=examples,
+    cache_examples=True,
+    theme='huggingface',
+)
 demo_app.launch(debug=True, enable_queue=True)
+demo_app_video.launch(debug=True, enable_queue=True)