update gradio for new models
Browse files
app.py
CHANGED
@@ -15,7 +15,8 @@ demo = gr.Interface(
|
|
15 |
fn=predict_fn,
|
16 |
inputs=[
|
17 |
gr.inputs.Image(label="Input Image"),
|
18 |
-
gr.Radio(['
|
|
|
19 |
],
|
20 |
outputs=[
|
21 |
gr.outputs.Label(label="Prediction"),
|
@@ -24,10 +25,10 @@ demo = gr.Interface(
|
|
24 |
title="Global Context Vision Transformer (GCViT) Demo",
|
25 |
description="Image Classification with GCViT Model using ImageNet Pretrain Weights.",
|
26 |
examples=[
|
27 |
-
["example/hot_air_ballon.jpg", '
|
28 |
-
["example/chelsea.png", '
|
29 |
-
["example/penguin.JPG", '
|
30 |
-
["example/bus.jpg", '
|
31 |
],
|
32 |
)
|
33 |
demo.launch()
|
|
|
15 |
fn=predict_fn,
|
16 |
inputs=[
|
17 |
gr.inputs.Image(label="Input Image"),
|
18 |
+
gr.Radio(['GCViTXXTiny', 'GCViTXTiny', 'GCViTTiny',
|
19 |
+
'GCViTSmall', 'GCViTBase','GCViTLarge'], value='GCViTXXTiny', label='Model Name')
|
20 |
],
|
21 |
outputs=[
|
22 |
gr.outputs.Label(label="Prediction"),
|
|
|
25 |
title="Global Context Vision Transformer (GCViT) Demo",
|
26 |
description="Image Classification with GCViT Model using ImageNet Pretrain Weights.",
|
27 |
examples=[
|
28 |
+
["example/hot_air_ballon.jpg", 'GCViTXXTiny'],
|
29 |
+
["example/chelsea.png", 'GCViTXXTiny'],
|
30 |
+
["example/penguin.JPG", 'GCViTXXTiny'],
|
31 |
+
["example/bus.jpg", 'GCViTXXTiny'],
|
32 |
],
|
33 |
)
|
34 |
demo.launch()
|