Update app.py

app.py

@@ -18,18 +18,18 @@ def generate_caption_clipgpt(image):
 
 # Sample image paths
 sample_images = [
-    "CXR191_IM-0591-1001.jpg",
-    "CXR191_IM-0598-1001.jpg",
-    "CXR191_IM-0601-1001.jpg",
-    "CXR191_IM-0609-1001.jpg",
-    "CXR191_IM-0618-1001.jpg"
+    "CXR191_IM-0591-1001.png",
+    "CXR191_IM-0598-1001.png",
+    "CXR191_IM-0601-1001.png",
+    "CXR191_IM-0609-1001.png",
+    "CXR191_IM-0618-1001.png"
 ]
 
 # Gradio interface 
 with gr.Blocks() as demo:
     with gr.Row():
         image = gr.Image(label="Upload Chest X-ray") 
-        # sample_images_gallery = gr.Gallery(value=sample_images, label="Sample Images", columns=5) 
+        sample_images_gallery = gr.Gallery(value=sample_images, label="Sample Images", columns=5) 
     with gr.Row():  
         model_choice = gr.Radio(["CLIP-GPT2", "ViT-GPT2", "ViT-CoAttention"], label="Select Model")
     with gr.Row():
@@ -44,6 +44,6 @@ with gr.Blocks() as demo:
 
     # Handle changes for both uploaded and sample images
     image.change(predict, [image, model_choice], caption)
-    # sample_images_gallery.change(predict, [sample_images_gallery, model_choice], caption)
+    sample_images_gallery.change(predict, [sample_images_gallery, model_choice], caption)
 
 demo.launch()