leuschnm commited on
Commit
49140ff
·
1 Parent(s): dc44445
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -59,7 +59,7 @@ def prepare_dataset(parameters, df, rain, temperature, datepicker):
59
  return df.to_dataloader(train=False, batch_size=256,num_workers = 0)
60
 
61
  @st.cache_data
62
- def predict(model, dataloader):
63
  out = model.predict(dataloader, mode="raw", return_x=True, return_index=True)
64
  preds = raw_preds_to_df(out, quantiles = None)
65
 
 
59
  return df.to_dataloader(train=False, batch_size=256,num_workers = 0)
60
 
61
  @st.cache_data
62
+ def predict(_model, dataloader):
63
  out = model.predict(dataloader, mode="raw", return_x=True, return_index=True)
64
  preds = raw_preds_to_df(out, quantiles = None)
65