Update app.py

app.py

@@ -31,20 +31,43 @@ sliders = [
         gr.Slider(minimum=3.75, maximum=5.52, step=0.1, label="Elongation"),
         gr.Slider(minimum=8.04, maximum=10.54, step=0.1, label="Elon_CV"),
 ]
-slider_rows = gr.Column([
-    gr.Row(sliders[0:3]),
-    gr.Row(sliders[3:6]),
-    gr.Row(sliders[6:9]),
-    gr.Row(sliders[9:12]),
-])
+
+with gr.Blocks() as demo:
+    gr.Markdown("## RPM Prediction App")
+    gr.Markdown("Adjust sliders for 12 features to predict RPM using a Decision Tree Regressor.")
+
+    with gr.Row():
+        sliders_row1 = sliders[0:4]
+        for slider in sliders_row1:
+            slider.render()
+
+    with gr.Row():
+        sliders_row2 = sliders[4:8]
+        for slider in sliders_row2:
+            slider.render()
+
+    with gr.Row():
+        sliders_row3 = sliders[8:12]
+        for slider in sliders_row3:
+            slider.render()
+
+    output = gr.Number(label="Predicted RPM")
+    predict_btn = gr.Button("Predict RPM")
+
+    predict_btn.click(
+        fn=predict_rpm,
+        inputs=sliders,
+        outputs=output
+    )
+
 # Create the interface
-demo = gr.Interface(
-    fn=predict_rpm,
-    inputs=sliders,
-    outputs=gr.Number(label="Predicted RPM"),
-    title="RPM Prediction App",
-    description="Adjust the sliders for each of the 12 features to predict the RPM."
-)
+# demo = gr.Interface(
+#     fn=predict_rpm,
+#     inputs=sliders,
+#     outputs=gr.Number(label="Predicted RPM"),
+#     title="RPM Prediction App",
+#     description="Adjust the sliders for each of the 12 features to predict the RPM."
+# )
 
 if __name__ == "__main__":
     demo.launch()