fix for base and small model

app.py

@@ -15,14 +15,14 @@ demo = gr.Interface(
     fn=predict_fn,
     inputs=[
         gr.inputs.Image(label="Input Image"),
-        gr.Radio(['GCViTTiny', 'GCViTSmall', 'GCViTBase'], value='GCViTTiny', label='Model Size')
+        gr.Radio(['GCViTTiny', 'GCViTSmall', 'GCViTBase'], value='GCViTTiny', label='Model Name')
         ],
     outputs=[
         gr.outputs.Label(label="Prediction"),
         gr.inputs.Image(label="GradCAM"),
     ],
     title="Global Context Vision Transformer (GCViT) Demo",
-    description="Image Classification with ImageNet Pretrain Weights.",
+    description="Image Classification with GCViT Model using ImageNet Pretrain Weights.",
     examples=[
         ["example/hot_air_ballon.jpg", 'GCViTTiny'],
         ["example/chelsea.png", 'GCViTTiny'],

gcvit/models/gcvit.py

@@ -5,7 +5,7 @@ from ..layers import PatchEmbed, GCViTLayer, Identity
 
 
 BASE_URL = 'https://github.com/awsaf49/gcvit-tf/releases/download'
-TAG = 'v1.0.0'
+TAG = 'v1.0.4'
 NAME2CONFIG = {
     'gcvit_tiny': {'window_size': (7, 7, 14, 7),
                     'dim': 64,