Update app.py

app.py

@@ -53,8 +53,11 @@ def load_all_models():
 
     return nhits_models, timesnet_models, lstm_models, tft_models
 
-def generate_forecast(model, df):
-    forecast_df = model.predict(df=df)
+def generate_forecast(model, df,tag=False):
+    if tag:
+        forecast_df = model.predict()
+    else:
+        forecast_df = model.predict(df=df)
     return forecast_df
 
 def determine_frequency(df, ds_col):
@@ -171,7 +174,7 @@ def forecast_time_series(df, model_type, horizon, max_steps=200, ds_col='ds'):
     
     forecast_results = {}
     st.sidebar.write(f"Generating forecast using {model_type} model...")
-    forecast_results[model_type] = generate_forecast(model, df)
+    forecast_results[model_type] = generate_forecast(model, df, tag='retrain')
 
     for model_name, forecast_df in forecast_results.items():
         plot_forecasts(forecast_df, df, f'{model_name} Forecast Comparison')