change conversion

app.py

@@ -55,9 +55,9 @@ def prepare_dataset(_parameters, df, rain, temperature, datepicker, mapping):
     df = TimeSeriesDataSet.from_parameters(_parameters, df)
     return df.to_dataloader(train=False, batch_size=256,num_workers = 0)
 
-def predict(_model, _dataloader):
-    out = _model.predict(_dataloader, mode="raw", return_x=True, return_index=True)#, trainer_kwargs=dict(accelerator="cpu"))
-    preds = raw_preds_to_df(out)
+def predict(model, dataloader):
+    out = model.predict(dataloader, mode="raw", return_x=True, return_index=True)#, trainer_kwargs=dict(accelerator="cpu"))
+    preds = raw_preds_to_df(out, quantiles = None)
     return preds[["pred_idx", "Group", "pred"]]
 
 def adjust_data_for_plot(df, preds):