Update app.py

app.py

@@ -39,7 +39,7 @@ 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,7)), label="Example Image Selected")
+        imgID = gr.Dropdown(["1","2","3","4"], 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,8 @@ 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(str(imgID))
+        print(imgID)
+        img = getImageID(imgID)
         return generate_caption_cnnrnn(img), getCaption(imgID)
     else:
         return "select a model","select an image"