Update app.py

app.py

@@ -186,19 +186,17 @@ with gr.Blocks() as demo:
             gr.Markdown("### Input Parameters")
             for inp in inputs:
                 inp.render()
+            # Submit and Clear Buttons under the inputs
+            gr.Button(value="Submit", variant="primary").click(
+                fn=predict_and_plot, 
+                inputs=inputs, 
+                outputs=[gr.Plot(label="Contamination Levels Over Time")] + contamination_outputs + gradients_outputs + cleaning_time_outputs
+            )
+            gr.Button(value="Clear").click(fn=lambda: None)
 
         with gr.Column(scale=1):
             gr.Image(image_path)
 
-    # Middle Section: Submit and Clear Buttons
-    with gr.Row():
-        gr.Button(value="Submit", variant="primary").click(
-            fn=predict_and_plot, 
-            inputs=inputs, 
-            outputs=[gr.Plot(label="Contamination Levels Over Time")] + contamination_outputs + gradients_outputs + cleaning_time_outputs
-        )
-        gr.Button(value="Clear").click(fn=lambda: None)
-
     # Bottom Section: Outputs (Three columns) and Plot Below
     with gr.Row():
         with gr.Column(scale=2):