Update app.py

app.py

@@ -91,7 +91,15 @@ with gr.Blocks() as demo:
 
     # Event handlers 
     generate_button.click(predict, [image, model_choice, max_tokens, temperature], caption) 
-    image_table.click(predict_from_table, [image_table, model_choice], outputs=[image_display, caption]) 
+
+    # Create an event handler for the entire Dataframe
+    def dataframe_selected(index, row, model_choice):
+        if row is not None:  # Check if an actual row was selected
+            return predict_from_table(row, model_choice)
+    
+    # Attach the function to the dataframe
+    image_table.change(dataframe_selected, inputs=[image_table, model_choice], outputs=[image_display, caption])
+    
 
 
 demo.launch()