Update app.py

app.py

@@ -4,6 +4,7 @@ from statsmodels.tsa.holtwinters import ExponentialSmoothing
 import numpy as np
 from itertools import product
 from io import BytesIO
+import plotly.express as px
 
 # Function to run the Exponential Smoothing Model
 def run_exp_smoothing(city_data, trend, damped_trend, seasonal, seasonal_period):
@@ -68,7 +69,8 @@ if forecast is not None:
     # Ensure the index is correctly formatted as 'YYYY-MM'
     # forecast_df.index = forecast_df.index.strftime('%Y-%m')
     st.table(forecast_df)
-    st.plotly_chart(forecast_df)
+    fig = px.line(forecast_df, x=forecast_df.index, y="Forecast")
+    st.plotly_chart(fig)
 # ... [rest of the code]