Update app.py

app.py

@@ -136,4 +136,4 @@ if st.button('Train Models'):
         y_gru_pred = scaler.inverse_transform(y_gru_pred)
     
         # Visualization
-        plot_predictions(train_data, test_data, y_rnn_pred, y_lstm_pred, y_gru_pred)
+        plot_predictions(train_data, test_data, y_rnn_pred, y_lstm_pred, y_gru_pred, look_back)