latif98 commited on
Commit
13ed778
1 Parent(s): 7934763

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -160,8 +160,8 @@ model = VideoMAEForVideoClassification.from_pretrained(model_name)
160
 
161
  demo = gr.Interface(
162
  run_inference,
163
- [gr.Video(), model, image_processor],
164
- gr.Label(num_top_classes=1, interpret_by_tokens=True),
165
  title="VideoMAE fine-tuned on numbers, alphabets and nouns videos.",
166
  description="Gradio demo app of fine-tuned VideoMAE for video classification, To use it simply upload your video.",
167
  article="VideoMAE"
 
160
 
161
  demo = gr.Interface(
162
  run_inference,
163
+ [model,gr.Video(), image_processor],
164
+ outputs = 'text',
165
  title="VideoMAE fine-tuned on numbers, alphabets and nouns videos.",
166
  description="Gradio demo app of fine-tuned VideoMAE for video classification, To use it simply upload your video.",
167
  article="VideoMAE"