Update app.py
Browse files
app.py
CHANGED
|
@@ -146,32 +146,32 @@ def inference_image(img):
|
|
| 146 |
return {imagenet_id_to_classname[str(i)]: float(prediction[i]) for i in range(1000)}
|
| 147 |
|
| 148 |
|
| 149 |
-
demo = gr.Interface(
|
| 150 |
-
|
| 151 |
-
|
| 152 |
-
|
| 153 |
|
| 154 |
-
)
|
| 155 |
-
|
| 156 |
-
|
| 157 |
-
|
| 158 |
-
|
| 159 |
-
#
|
| 160 |
-
|
| 161 |
-
|
| 162 |
-
|
| 163 |
-
|
| 164 |
-
#
|
| 165 |
-
# #
|
| 166 |
-
|
| 167 |
-
|
| 168 |
-
|
| 169 |
-
|
| 170 |
-
#
|
| 171 |
-
|
| 172 |
-
|
| 173 |
-
|
| 174 |
-
|
| 175 |
# with gr.Row():
|
| 176 |
# gr.Examples(examples=['./videos/hitting_baseball.mp4', './videos/hoverboarding.mp4', './videos/yoga.mp4'], inputs=input_video, outputs=label_video, fn=inference_video, cache_examples=True)
|
| 177 |
|
|
@@ -194,7 +194,7 @@ demo = gr.Interface(
|
|
| 194 |
# """
|
| 195 |
# )
|
| 196 |
|
| 197 |
-
|
| 198 |
# # submit_image_button.click(fn=inference_image, inputs=input_image, outputs=label_image)
|
| 199 |
|
| 200 |
demo.launch()
|
|
|
|
| 146 |
return {imagenet_id_to_classname[str(i)]: float(prediction[i]) for i in range(1000)}
|
| 147 |
|
| 148 |
|
| 149 |
+
# demo = gr.Interface(
|
| 150 |
+
# fn = ultra_inference_video,
|
| 151 |
+
# inputs = "sketchpad",
|
| 152 |
+
# outputs = "label",
|
| 153 |
|
| 154 |
+
# )
|
| 155 |
+
demo = gr.Blocks()
|
| 156 |
+
with demo:
|
| 157 |
+
gr.Markdown(
|
| 158 |
+
"""
|
| 159 |
+
# VideoMamba-Ti
|
| 160 |
+
Gradio demo for <a href='https://github.com/OpenGVLab/VideoMamba' target='_blank'>VideoMamba</a>: To use it, simply upload your video, or click one of the examples to load them. Read more at the links below.
|
| 161 |
+
"""
|
| 162 |
+
)
|
| 163 |
+
|
| 164 |
+
# with gr.Tab("Video"):
|
| 165 |
+
# # with gr.Box():
|
| 166 |
+
with gr.Row():
|
| 167 |
+
with gr.Column():
|
| 168 |
+
with gr.Row():
|
| 169 |
+
input_video = gr.Video(label='Input Video', height=360)
|
| 170 |
+
# input_video = load_video(input_video)
|
| 171 |
+
with gr.Row():
|
| 172 |
+
submit_video_button = gr.Button('Submit')
|
| 173 |
+
with gr.Column():
|
| 174 |
+
label_video = gr.Label(num_top_classes=10)
|
| 175 |
# with gr.Row():
|
| 176 |
# gr.Examples(examples=['./videos/hitting_baseball.mp4', './videos/hoverboarding.mp4', './videos/yoga.mp4'], inputs=input_video, outputs=label_video, fn=inference_video, cache_examples=True)
|
| 177 |
|
|
|
|
| 194 |
# """
|
| 195 |
# )
|
| 196 |
|
| 197 |
+
submit_video_button.click(fn=inference_video, inputs=input_video, outputs=label_video)
|
| 198 |
# # submit_image_button.click(fn=inference_image, inputs=input_image, outputs=label_image)
|
| 199 |
|
| 200 |
demo.launch()
|