Spaces:
Runtime error
Runtime error
File size: 715 Bytes
59d3dd1 0f220e9 c6939df 59d3dd1 0f220e9 c6939df 0f220e9 c6939df 0f220e9 59d3dd1 c6939df 0f220e9 c6939df 0f220e9 dc386c1 c6939df |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 |
import gradio as gr
import utils
from config import KINETICS_600_LABELS, MODEL
from logger import logging
def get_predictions(video_path):
logging.info(f">>> Getting predictions for video file : {video_path}")
video, _ = utils.preprocess_video(video_path)
model = MODEL
probs = model(video)
labels = utils.get_top_k(probs, label_map=KINETICS_600_LABELS)
logging.info(f"Getting predictions successful : {labels}")
return labels
label = gr.components.Label(num_top_classes=5)
vd = gr.components.Video()
logging.info(">>> Launching the gradio app... ")
iface = gr.Interface(fn=get_predictions, inputs=vd, outputs=label)
iface.launch()
logging.info(">>> Launched successfully.")
|