Update app.py

app.py

@@ -29,7 +29,7 @@ sample_images = [
 with gr.Blocks() as demo:
     with gr.Row():
         image = gr.Image(label="Upload Chest X-ray")
-        # sample_image_gallery = gr.ImageGallery(sample_images, label="Sample Images")
+        sample_image_gallery = gr.ImageGallery(sample_images, label="Sample Images")
     with gr.Row():  
         model_choice = gr.Radio(["CLIP-GPT2", "ViT-GPT2", "ViT-CoAttention"], label="Select Model")
     with gr.Row():
@@ -43,7 +43,7 @@ with gr.Blocks() as demo:
             return "Caption generation for this model is not yet implemented."
 
     # Handle changes for both uploaded and sample images
-    gr.Image.change(predict, [image, model_choice], caption)
-    # sample_image_gallery.change(predict, [sample_image_gallery, model_choice], caption)
+    image.change(predict, [image, model_choice], caption)
+    sample_image_gallery.change(predict, [sample_image_gallery, model_choice], caption)
 
 demo.launch()