Hotfix - add button for Aaron

jvivian · Jun 10, 2024 · 72421c2 · 72421c2
1 parent d5d3816
commit 72421c2
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/covid19_drdfm/streamlit/pages/0_Dynamic_Factor_Model.py b/covid19_drdfm/streamlit/pages/0_Dynamic_Factor_Model.py
@@ -173,6 +173,10 @@ def run_model(ad, out_dir, batch, global_multiplier) -> ModelRunner:
 ad = data.ad
 global_multiplier, out_dir = additional_params()
 batch = None if ad.obs.empty else ad.obs.columns[0]
+
+if not st.button("Run Model"):
+    st.stop()
+
 dfm = run_model(ad, out_dir, batch, global_multiplier)
 st.balloons()