Update app.py

app.py

@@ -81,8 +81,7 @@ def inference(user_image=None, selected_example=None):
         input_image = process_image(user_image)
     elif selected_example is not None:
         print(selected_example)
-        example_image = load_example(selected_example)
-        input_image = process_image(example_image)
+        input_image = process_image(selected_example)
     else:
         return "Please upload an image or select an example image."
 
@@ -142,22 +141,28 @@ with gr.Blocks() as demo:
     )
     with gr.Row():
         with gr.Column():
-            inp_image = gr.Image(label="Upload Image")
-        with gr.Column():
-            example_selection = gr.Gallery(
-                value=list(example_images.values()),
-                label="Click an Example Image",
+            inp_image = gr.Image(label="Upload Image", type="pil")
+            example_selection = gr.Radio(
+                choices=list(example_images.keys()),
+                label="Select Example Image",
                 type="value"
             )
-
     with gr.Column():
         output = gr.Markdown()
 
+    # Handle user image input
+    inp_image.change(handle_inputs, inputs=[inp_image, example_selection], outputs=output)
+
+    # Handle example image selection
+    example_selection.change(
+        handle_inputs,
+        inputs=[inp_image, example_selection],  # This allows example image selection without needing to upload
+        outputs=output
+    )
+
     def handle_inputs(user_image, selected_example):
         return inference(user_image=user_image, selected_example=selected_example)
 
-    inp_image.change(handle_inputs, inputs=[inp_image, example_selection], outputs=output)
-    example_selection.change(handle_inputs, inputs=[inp_image, example_selection], outputs=output)
 
 if __name__ == "__main__":
     demo.launch()