Update app.py

app.py

@@ -33,28 +33,35 @@ with gr.Blocks() as demo:
     gr.HTML("<p style='text-align: center;'>You can generate captions by uploading an X-Ray and selecting a model of your choice below</p>")
 
        
-    with gr.Row():
-        sample_images = [
-        'https://imgur.com/W1pIr9b',
-        'https://imgur.com/MLJaWnf',
-        'https://imgur.com/6XymFW1',
-        'https://imgur.com/zdPjZZ1',
-        'https://imgur.com/DKUlZbF'
-            ]
-
-        
-        image = gr.Image(label="Upload Chest X-ray", type="pil")    
-        
-        sample_images_gallery = gr.Gallery(value = sample_images,label="Sample Images")
+    examples_list = [
+        ['https://imgur.com/W1pIr9b', 'Caption for image 1'],
+        ['https://imgur.com/MLJaWnf', 'Caption for image 2']
+    ]
+    examples = gr.Examples(examples_list, 
+                           inputs="image",  
+                           outputs=["image", "text"], 
+                           label="Example Images",
+                           fn=render_image  
+                       ) 
+
+    image_table = gr.Dataframe(headers=['image', 'caption'], datatype=['html', 'str'])  
+
 
     with gr.Row():
         model_choice = gr.Radio(["CLIP-GPT2", "ViT-GPT2", "ViT-CoAttention"], label="Select Model")
 
         generate_button = gr.Button("Generate Caption") 
     
+    caption = gr.Textbox(label="Generated Caption") 
+
     
+    # Function to populate table rows when an example image is clicked
+    def populate_table_from_example(image_path, caption):
+        new_row = [[render_image(image_path)], [caption]]
+        image_table.append(new_row) 
+
+    examples.click(populate_table_from_example, inputs=examples, outputs=image_table)
 
-    caption = gr.Textbox(label="Generated Caption") 
 
     def predict(img, model_name):
         if model_name == "CLIP-GPT2":