Spaces:
Runtime error
Runtime error
Commit ·
1bc2256
1
Parent(s): c09b2c5
Update app.py
Browse files
app.py
CHANGED
|
@@ -88,4 +88,26 @@ def model_interface(uploaded_video, model_choice, activity):
|
|
| 88 |
likely_label = activities_list[max_prob_index]
|
| 89 |
likely_probability = float(probs[0][max_prob_index]) * 100
|
| 90 |
|
| 91 |
-
return concatenated_image, results_probs, results_logits, [likely_label, likely_probability]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 88 |
likely_label = activities_list[max_prob_index]
|
| 89 |
likely_probability = float(probs[0][max_prob_index]) * 100
|
| 90 |
|
| 91 |
+
return concatenated_image, results_probs, results_logits, [likely_label, likely_probability]
|
| 92 |
+
|
| 93 |
+
iface = gr.Interface(
|
| 94 |
+
fn=model_interface,
|
| 95 |
+
inputs=[
|
| 96 |
+
gr.components.Video(label="Upload a video file"),
|
| 97 |
+
gr.components.Dropdown(choices=[
|
| 98 |
+
"microsoft/xclip-base-patch16-zero-shot",
|
| 99 |
+
"microsoft/xclip-base-patch32-16-frames",
|
| 100 |
+
"microsoft/xclip-base-patch32"
|
| 101 |
+
], label="Model Choice"),
|
| 102 |
+
gr.components.Textbox(default="dancing", label="Desired Activity to Recognize"),
|
| 103 |
+
],
|
| 104 |
+
outputs=[
|
| 105 |
+
gr.components.Image(type="pil", label="Sampled Frames"),
|
| 106 |
+
gr.components.Textbox(type="text", label="Probabilities"),
|
| 107 |
+
gr.components.Textbox(type="text", label="Raw Scores"),
|
| 108 |
+
gr.components.Textbox(type="text", label="Top Prediction")
|
| 109 |
+
],
|
| 110 |
+
live=False
|
| 111 |
+
)
|
| 112 |
+
|
| 113 |
+
iface.launch()
|