norfair-demo / app.py
Diego Fernandez
feat: initial version
fca2efd
raw
history blame
684 Bytes
import gradio as gr
from inference import inference
dd_model = gr.Dropdown(choices=["YoloV7"], value="YoloV7", label="Model")
cb_motion_estimation = gr.Checkbox(value=True, label="Motion estimation")
cb_path_draw = gr.Checkbox(value=True, label="Drawing paths")
dd_track_points = gr.Dropdown(
choices=["Boxes", "Centroid"], value="Boxes", label="Detections style"
)
slide_threshold = gr.Slider(minimum=0, maximum=1, value=0.25, label="Model confidence threshold")
inputs = ["video", dd_model, cb_motion_estimation, cb_path_draw, dd_track_points, slide_threshold]
outputs = "playablevideo"
iface = gr.Interface(fn=inference, inputs=inputs, outputs=outputs)
iface.launch()