Spaces:
Runtime error
Runtime error
bug fix
Browse files
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(
|
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 |
|