Update app.py

app.py

@@ -124,9 +124,15 @@ with gr.Blocks() as demo:
 
     # Toggle input fields based on model selection
     def toggle_inputs(choice):
-        return gr.update(visible=(choice == "ViT")), gr.update(visible=(choice == "Neural Network"))
+    """Toggle visibility of inputs based on model selection."""
+    image_visibility = choice == "ViT"
+    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])
+
     example_btn_1.click(lambda: fill_example(benign_example), None, feature_inputs)
     example_btn_2.click(lambda: fill_example(malignant_example), None, feature_inputs)