Update app.py

app.py

@@ -198,25 +198,19 @@ with gr.Blocks() as demo:
             )
             gr.Button(value="Clear").click(fn=lambda: None)
             
-    # Plot above the three columns of outputs
-    with gr.Column():
-        gr.Markdown("### Predictions")
-        gr.Plot(label="Contamination Levels Over Time").render()
-
-        with gr.Row():
-            with gr.Column():
-                gr.Markdown("#### Contamination Predictions")
-                for out in contamination_outputs:
-                    out.render()
+        with gr.Column():
+            gr.Markdown("### Contamination Predictions")
+            for out in contamination_outputs:
+                out.render()
                 
-            with gr.Column():
-                gr.Markdown("#### Gradient Predictions")
-                for out in gradients_outputs:
-                    out.render()
-
-            with gr.Column():
-                gr.Markdown("#### Cleaning Time Predictions")
-                for out in cleaning_time_outputs:
-                    out.render()
+        with gr.Column():
+            gr.Markdown("### Gradient Predictions")
+            for out in gradients_outputs:
+                out.render()
+
+        with gr.Column():
+            gr.Markdown("### Cleaning Time Predictions")
+            for out in cleaning_time_outputs:
+                out.render()
 
 demo.launch()