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(["1","2","3","4"], label="Choose the ID of the image selected")
+        imgID = gr.Dropdown(["1","2","3","4","5","6"], label="Choose the ID of the 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")