IbrahimHasani commited on
Commit
1bc2256
·
1 Parent(s): c09b2c5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +23 -1
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()