change conversion

app.py

@@ -57,7 +57,7 @@ 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, datepicker):
+def predict(_model, _dataloader, datepicker, df):
     out = _model.predict(_dataloader, mode="raw", return_x=True, return_index=True)
     first_date = min(df["Date"])
     preds = raw_preds_to_df(out, first_date)
@@ -157,7 +157,7 @@ def main():
     rain = st.selectbox("Rain Indicator", ('Default', 'Yes', 'No'), key = "rain")
 
     dataloader = prepare_dataset(parameters, df.copy(), st.session_state.rain, st.session_state.temperature, st.session_state.date, RAIN_MAPPING)
-    preds = predict(model, dataloader, st.session_state.date)
+    preds = predict(model, dataloader, st.session_state.date, df)
 
     data_plot = adjust_data_for_plot(df.copy(), preds)
     fig, _ = generate_plot(data_plot)