Update app.py

app.py

@@ -88,12 +88,27 @@ def classify(model_choice, image=None, *features):
 # Gradio UI
 model_selector = gr.Radio(["ViT", "Neural Network"], label="Choose Model")
 image_input = gr.Image(type="pil", label="Upload Mammogram Image")
-feature_inputs = [gr.Number(label=feature) for feature in feature_names]
+
+# feature_inputs = [gr.Number(label=feature, scale=0.5) for feature in feature_names]
+num_columns = 3  # Change to 4 for a 4-column layout
+
+feature_inputs = []
+with gr.Blocks():  # Ensures proper layout management
+    for i in range(0, len(feature_names), num_columns):
+        with gr.Row():
+            for feature in feature_names[i:i+num_columns]:
+                feature_inputs.append(gr.Number(label=feature, scale=0.5))
+
 
 # Example buttons
 def fill_example(example):
     return {feature_inputs[i]: example[i] for i in range(len(feature_inputs))}
 
+examples = [
+    ["Neural Network", None] + benign_example,
+    ["Neural Network", None] + malignant_example
+]
+
 iface = gr.Interface(
     fn=classify,
     inputs=[model_selector, image_input] + feature_inputs,