Update app.py
Browse files
app.py
CHANGED
@@ -161,9 +161,7 @@ model = VideoMAEForVideoClassification.from_pretrained(model_name)
|
|
161 |
demo = gr.Interface(
|
162 |
run_inference,
|
163 |
[gr.Video(), model, image_processor],
|
164 |
-
|
165 |
-
interpretation="default",
|
166 |
-
examples=[["path/to/example/video.mp4"]],
|
167 |
title="VideoMAE fine-tuned on numbers, alphabets and nouns videos.",
|
168 |
description="Gradio demo app of fine-tuned VideoMAE for video classification, To use it simply upload your video.",
|
169 |
article="VideoMAE"
|
|
|
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"
|