bug fix

app.py

@@ -54,8 +54,8 @@ def prepare_dataset(parameters, df, rain, temperature, datepicker, mapping):
     return df.to_dataloader(train=False, batch_size=256,num_workers = 0)
 
 @st.cache_data
-def predict(_model, _dataloader):
-    out = model.predict(dataloader, mode="raw", return_x=True, return_index=True)
+def predict(_model, _dataloader, datepicker):
+    out = _model.predict(_dataloader, mode="raw", return_x=True, return_index=True)
     preds = raw_preds_to_df(out, quantiles = None)
     
     date_list = [datepicker + datetime.timedelta(days=x) for x in range(30)]
@@ -146,7 +146,7 @@ def main():
 
     if st.button("Forecast Sales", type="primary"):
         dataloader = prepare_dataset(parameters, df, rain, temperature, datepicker, rain_mapping)
-        preds = predict(model, dataloader)
+        preds = predict(model, dataloader, datepicker)
         generate_plot(df, preds)
 
 if __name__ == '__main__':