Spaces:
Runtime error
Runtime error
change conversion
Browse files
app.py
CHANGED
@@ -56,7 +56,7 @@ def prepare_dataset(_parameters, df, rain, temperature, datepicker, mapping):
|
|
56 |
return df.to_dataloader(train=False, batch_size=256,num_workers = 0)
|
57 |
|
58 |
def predict(_model, _dataloader):
|
59 |
-
out = _model.predict(_dataloader, mode="raw", return_x=True,return_index=True
|
60 |
preds = raw_preds_to_df(out)
|
61 |
return preds[["pred_idx", "Group", "pred"]]
|
62 |
|
|
|
56 |
return df.to_dataloader(train=False, batch_size=256,num_workers = 0)
|
57 |
|
58 |
def predict(_model, _dataloader):
|
59 |
+
out = _model.predict(_dataloader, mode="raw", return_x=True, return_index=True)#, trainer_kwargs=dict(accelerator="cpu"))
|
60 |
preds = raw_preds_to_df(out)
|
61 |
return preds[["pred_idx", "Group", "pred"]]
|
62 |
|