awsaf49 commited on
Commit
28f10f6
·
1 Parent(s): 8778bc5

update gradio for new models

Browse files
Files changed (1) hide show
  1. app.py +6 -5
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(['GCViTTiny', 'GCViTSmall', 'GCViTBase'], value='GCViTTiny', label='Model Name')
 
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", 'GCViTTiny'],
28
- ["example/chelsea.png", 'GCViTTiny'],
29
- ["example/penguin.JPG", 'GCViTTiny'],
30
- ["example/bus.jpg", 'GCViTTiny'],
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()