Update app.py

app.py

@@ -32,14 +32,14 @@ def generate_caption_cnnrnn(image):
 
 
 with gr.Row():
-    image = gr.Image(label="Upload Chest X-ray", type="pil", height='100',width='100')   
+    image = gr.Image(label="Upload Chest X-ray", type="pil", height='50',width='50')   
                      
  
 with gr.Row():
     with gr.Column(): # Column for dropdowns and model choice
         max_tokens = gr.Dropdown(list(range(50, 101)), label="Max Tokens", value=75)
         temperature = gr.Slider(0.5, 0.9, step=0.1, label="Temperature", value=0.9)
-        imgID = gr.Dropdown(list(range(1,6)), label="Example Image Selected", value=1)
+        imgID = gr.Dropdown(list(range(1,7)), label="Example Image Selected")
 
     model_choice = gr.Radio(["CLIP-GPT2", "ViT-GPT2", "ViT-CoAttention", "Baseline Model CNN-RNN"], label="Select Model") 
     generate_button = gr.Button("Generate Caption") 
@@ -67,7 +67,7 @@ def predict(img, model_name, max_tokens, temperature, imgID):
     elif model_name == "ViT-CoAttention":
         return generate_caption_vitCoAtt(img),  getCaption(imgID)
     elif model_name == "Baseline Model CNN-RNN":
-        img = getImageID(imgID)
+        img = getImageID(str(imgID))
         return generate_caption_cnnrnn(img), getCaption(imgID)
     else:
         return "select a model","select an image"