Update app.py
Browse files
app.py
CHANGED
@@ -160,8 +160,8 @@ model = VideoMAEForVideoClassification.from_pretrained(model_name)
|
|
160 |
|
161 |
demo = gr.Interface(
|
162 |
run_inference,
|
163 |
-
[gr.Video(),
|
164 |
-
|
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"
|