Update app.py

app.py

@@ -130,33 +130,43 @@ def load_example(example_name):
     if image_path:
         return Image.open(image_path)
     return None
-    
+           
 with gr.Blocks() as demo:
-    gr.Markdown(
-        """
+    gr.Markdown("""
         # Image to Video App
         Find your favorite Gordon Ramasay scene by uploading an image from the scene, the app will thereafter find a corresponding youtube video for that scene. 
         Or try one of our examples (unseen data for the model).
-        """
-    )
+        """)
+
     with gr.Row():
         with gr.Column():
             inp_image = gr.Image(label="Upload Image", type="pil")
             example_selection = gr.Radio(
-                choices=list(example_images.values()),
+                choices=list(example_images.keys()),
                 label="Select Example Image",
-                type="value"
+                type="value"  # Ensure single string return value
             )
-    with gr.Column():
-        output = gr.Markdown()
+            example_display = gr.Image(label="Selected Example Image", type="pil")
+
+        with gr.Column():
+            output = gr.Markdown()
 
+    # Update the example image when an example is selected
+    example_selection.change(
+        lambda selected_example: load_example(selected_example),
+        inputs=[example_selection],
+        outputs=[example_display]
+    )
+
+    # Submit button to trigger inference
     submit_button = gr.Button("Submit")
 
+    # Handle submit button click
     submit_button.click(
         lambda user_image, selected_example: inference(user_image=user_image, selected_example=selected_example),
         inputs=[inp_image, example_selection],
         outputs=output
     )
-    
+
 if __name__ == "__main__":
     demo.launch()