leuschnm commited on
Commit
c25a5e4
·
1 Parent(s): be5d067

change conversion

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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, trainer_kwargs=dict(accelerator="cpu"))
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