Update app.py

app.py

@@ -158,7 +158,7 @@ def select_model(horizon, model_type, max_steps=200):
     else:
         raise ValueError(f"Unsupported model type: {model_type}")
 
-def model_train(model):
+def model_train(df,model):
     model.fit(df)
     return model
 def forecast_time_series(df, model_type, horizon, max_steps=200):
@@ -167,8 +167,10 @@ def forecast_time_series(df, model_type, horizon, max_steps=200):
     st.write(f"Determined frequency: {freq}")
     df['ds'] = pd.to_datetime(df['ds'], errors='coerce')
     df = df.dropna(subset=['ds'])
+    
     selected_model = select_model(horizon, model_type, max_steps)
-    model = model_train(selected_model)
+    model = model_train(df,selected_model)
+    
     forecast_results = {}
     st.write(f"Generating forecast using {model_type} model...")
     forecast_results[model_type] = generate_forecast(model, df)