SignERT-PoseOnlyDemo / src /simple_demo.py
CristianLazoQuispe's picture
demo v1
979a0f3
raw
history blame contribute delete
926 Bytes
import gradio as gr
from predict import predict_from_video
def classify_video(video):
top1, top5_df = predict_from_video(video)
return f"Top-1: {top1}", top5_df
demo = gr.Interface(
fn=classify_video,
inputs=gr.Video(sources=["upload", "webcam"], label="🎥 Video (webcam o archivo)"),
outputs=[
gr.Text(label="Predicción principal"),
gr.Label(num_top_classes=5),
],
#outputs="text",
title="🧠 ISLR Demo con Mediapipe y 100 Clases",
description="Clasificador de lenguaje de señas aislado. Muestra las Top-5 clases más probables.",
#examples=["/home/va0831/slr/SLR_2024/Gloss/SignLanguageRecognition/J7tP98oDxqE_000000_000066-msasl-book.mp4",
# "/home/va0831/slr/SLR_2024/Gloss/SignLanguageRecognition/J7tP98oDxqE_000000_000066-msasl-book.mp4"],
#example_labels = ["book","house"],
)
if __name__ == "__main__":
demo.launch(server_port=9090)