norfair-demo / app.py
diegofernandezc's picture
Enable caching
0ce37c1
raw
history blame
1.11 kB
import os
import gradio as gr
from inference import inference
input_video = gr.Video(mirror_webcam=False)
dd_model = gr.Dropdown(choices=["YOLOv7", "YOLOv7 Tiny"], value="YOLOv7", label="Model")
features = gr.CheckboxGroup(
choices=["Track camera movement", "Draw objects paths"],
value=["Track camera movement", "Draw objects paths"],
label="Features",
type="index",
)
cb_path_draw = gr.Checkbox(value=True, label="Draw objects paths")
dd_track_points = gr.Dropdown(
choices=["Bounding box", "Centroid"], value="Bounding box", label="Detections style"
)
slide_threshold = gr.Slider(minimum=0, maximum=1, value=0.25, label="Model confidence threshold")
intput_components = [
input_video,
dd_model,
features,
dd_track_points,
slide_threshold
]
output_components = "playablevideo"
example_list = [["examples/" + example] for example in os.listdir("examples")]
iface = gr.Interface(
fn=inference,
inputs=intput_components,
outputs=output_components,
examples=example_list,
cache_examples=True,
)
iface.launch()