Update app.py

app.py

@@ -32,10 +32,14 @@ with gr.Blocks() as demo:
         "CXR195_IM-0618-1001.png"
     ]
         image = gr.Image(label="Upload Chest X-ray")
-        sample_images_gallery = gr.Gallery(
+        gr.Gallery(
             value = sample_images,
             label="Sample Images",
         )
+        # sample_images_gallery = gr.Gallery(
+        #     value = sample_images,
+        #     label="Sample Images",
+        # )
     with gr.Row():
         model_choice = gr.Radio(["CLIP-GPT2", "ViT-GPT2", "ViT-CoAttention"], label="Select Model")
         generate_button = gr.Button("Generate Caption") 
@@ -48,7 +52,7 @@ with gr.Blocks() as demo:
             return "Caption generation for this model is not yet implemented."
 
     generate_button.click(predict, [image, model_choice], caption)  # Trigger prediction on button click 
-    sample_images_gallery.change(predict, [sample_images_gallery, model_choice], caption)  # Handle sample images
+    # sample_images_gallery.change(predict, [sample_images_gallery, model_choice], caption)  # Handle sample images
 
 
 demo.launch()