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()
|