Update app.py

app.py

@@ -177,8 +177,12 @@ with gr.Blocks() as demo:
         feature_visibility = choice == "Neural Network"
         return [gr.update(visible=image_visibility)] + [gr.update(visible=feature_visibility)] * len(feature_inputs)
 
-    model_selector.change(toggle_inputs, model_selector, [image_input, *feature_inputs])
-
+    # model_selector.change(toggle_inputs, model_selector, [image_input, *feature_inputs])
+    model_selector.change(toggle_inputs, model_selector, [image_input, *feature_inputs, file_input])
+    # Bind image preview buttons to update image_input
+    for btn, img_path in example_buttons:
+        btn.click(lambda p=img_path: Image.open(p), outputs=image_input)
+        
     example_btn_1.click(lambda: fill_example(benign_example), None, feature_inputs)
     example_btn_2.click(lambda: fill_example(malignant_example), None, feature_inputs)