Update app.py

app.py

@@ -4,7 +4,7 @@ import pandas as pd
 import plotly.express as px
 import plotly.graph_objs as go
 from statsforecast import StatsForecast
-from statsforecast.models import seasonal_naive, auto_arima
+from statsforecast.models import auto_arima
 from utils import calc_seasonality
 
 @st.cache
@@ -43,7 +43,7 @@ if uploaded_file is not None:
           with st.spinner('Wait for it...'):
               fcst = StatsForecast(
                                    series_train, 
-                                   models=[(auto_arima, seasonality), (seasonal_naive, seasonality)], 
+                                   models=[(auto_arima, seasonality), ], 
                                    freq=freq, 
                                    n_jobs=1
                                   )
@@ -80,16 +80,7 @@ if uploaded_file is not None:
                            line=dict(width=1),
                            showlegend=True
                        ),
-                       go.Scatter(
-                           name=f'seasonal_naive_season_length-{seasonality}',
-                           x=forecasts['ds'],
-                           y=forecasts[f'seasonal_naive_season_length-{seasonality}'],
-                           mode='lines',
-                           marker=dict(color='red'),
-                           line=dict(width=1),
-                           showlegend=True
-                       )
-                   ])
+          ])
           fig_forecast.update_layout(
                        yaxis_title=series,
                        title=f'Forecasting {series}'