Update app.py

app.py

@@ -84,7 +84,7 @@ def predict_and_plot(velocity, temperature, precipitation, humidity):
         contamination_levels, gradients = predict_contamination_and_gradients(example_data_scaled)
 
         # Simulate contamination levels at multiple time intervals
-        time_intervals = np.arange(0, 601, 60)  # Simulating time intervals from 0 to 600 seconds
+        time_intervals = np.arange(0, 3601, 60)  # Simulating time intervals from 0 to 600 seconds
 
         # Generate simulated contamination levels (linear interpolation between predicted values)
         simulated_contamination_levels = np.array([
@@ -189,7 +189,7 @@ with gr.Blocks() as demo:
             gr.Button(value="Submit", variant="primary").click(
             fn=predict_and_plot, 
             inputs=inputs, 
-            outputs= contamination_outputs + gradients_outputs + cleaning_time_outputs)
+            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):