Update app.py

app.py

@@ -28,13 +28,12 @@ sample_images = [
 # Gradio interface 
 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")
+        image = gr.Image(label="Upload Chest X-ray") 
+        sample_images = gr.Images(sample_images, label="Sample Images") # Updated Component Name
     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)