Update app.py

app.py

@@ -165,6 +165,9 @@ if st.button('Train Models'):
         st.write(f"Std is: {np.std(next_point_forecasts)}")
 
         # Create MC Simulation
-        mean_return = np.mean(next_point_forecasts)
-        std_dev = np.std(next_point_forecasts)
+        closing_prices = data['Close']
+        last_close = closing_prices.iloc[-1]
+        all_rets = next_point_forecasts / last_close - 1
+        mean_return = np.mean(all_rets)
+        std_dev = np.std(all_rets)
         plot_monte_carlo_forecasts(data=data, n_futures=n_futures, n_samples=n_samples, mean_return=mean_return, std_dev=std_dev)