bug fix

app.py

@@ -59,7 +59,8 @@ def predict(_model, _dataloader, datepicker):
     preds = raw_preds_to_df(out, quantiles = None)
 
     def add_dates(group, datepicker):
-        return group["date_imputed"] = [datepicker + datetime.timedelta(days=x) for x in range(30)]
+        group["date_imputed"] = [datepicker + datetime.timedelta(days=x) for x in range(30)]
+        return group
     
     preds["date_imputed"] = preds.groupby("Group").transform(add_dates)