Update app.py

app.py

@@ -36,7 +36,7 @@ head_height = headrace_level - tailrace_level
 st.write(f"Effective Head Height: {head_height:.2f} m")
 
 # Prepare input data
-input_data = np.array([[flow_rate, head_height, EFFICIENCY, plant_capacity]], dtype=np.float32)
+input_data = np.array([[flow_rate, head_height, EFFICIENCY]], dtype=np.float32)
 
 # Predict energy output using ONNX model
 if st.button("Predict Energy Output (Local ONNX Model)"):
@@ -70,7 +70,7 @@ if st.button("Generate Prediction Graph"):
     hourly_predictions = []
 
     for hourly_flow in river_flows:
-        hourly_input = np.array([[hourly_flow, head_height, EFFICIENCY, plant_capacity]], dtype=np.float32)
+        hourly_input = np.array([[hourly_flow, head_height, EFFICIENCY]], dtype=np.float32)
         hourly_result = onnx_session.run(None, {input_name: hourly_input})
         hourly_predictions.append(float(hourly_result[0][0]))