Update app.py

app.py

@@ -29,11 +29,12 @@ sample_images = [
 with gr.Blocks() as demo:
     with gr.Row():
         image = gr.Image(label="Upload Chest X-ray") 
-        sample_images = gr.Image(sample_images, label="Sample Images") # Updated Component Name
+        sample_images_gallery = gr.Gallery(sample_images, label="Sample Images") 
     with gr.Row():  
         model_choice = gr.Radio(["CLIP-GPT2", "ViT-GPT2", "ViT-CoAttention"], label="Select Model")
     with gr.Row():
         caption = gr.Textbox(label="Generated Caption") 
+
     def predict(img, model_name):
         if model_name == "CLIP-GPT2":
             return generate_caption_clipgpt(img)
@@ -43,6 +44,6 @@ with gr.Blocks() as demo:
 
     # Handle changes for both uploaded and sample images
     image.change(predict, [image, model_choice], caption)
-    sample_image_gallery.change(predict, [sample_image_gallery, model_choice], caption)
+    sample_images_gallery.change(predict, [sample_images_gallery, model_choice], caption)
 
 demo.launch()