Spaces:

leuschnm
/

TemporalFusionTransformer

Runtime error

leuschnm commited on Jul 3, 2023

Commit

5c6f44b

1 Parent(s): 83c5179

bug fix

Files changed (1) hide show

app.py CHANGED Viewed

@@ -156,7 +156,7 @@ def main():
         # otherwise set it to false
         st.session_state.pressed = False
-    if st.button("Forecast Sales", type="primary") or session.state.pressed:
         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)

         # otherwise set it to false
         st.session_state.pressed = False
+    if st.button("Forecast Sales", type="primary") or st.session_state.pressed:
         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)