feat: Add spinner

app.py

@@ -61,20 +61,25 @@ import os
 
 os.makedirs(prefix, exist_ok=True)
 
-denoised_anchor_to_anchor = predict_and_write(
-    model,
-    full_matrix_dir=HiCorr_data + f"/anchor_2_anchor.loop.{chromosome}",
-    input_name=HiCorr_data + f"/anchor_2_anchor.loop.{chromosome}.p_val",
-    outdir=prefix,
-    anchor_dir=anchors,
-    chromosome=chromosome,
-    small_matrix_size=128,
-    step_size=128,
-    dummy=5,
-    # max_dist,
-    val_cols=["obs", "exp", "pval"],
-    # keep_zeros,
-)
+denoised_anchor_to_anchor = None
+
+with st.spinner("Running the model"):
+    denoised_anchor_to_anchor = predict_and_write(
+        model,
+        full_matrix_dir=HiCorr_data + f"/anchor_2_anchor.loop.{chromosome}",
+        input_name=HiCorr_data + f"/anchor_2_anchor.loop.{chromosome}.p_val",
+        outdir=prefix,
+        anchor_dir=anchors,
+        chromosome=chromosome,
+        small_matrix_size=128,
+        step_size=128,
+        dummy=5,
+        # max_dist,
+        val_cols=["obs", "exp", "pval"],
+        # keep_zeros,
+    )
+
+st.success("Done!")
 
 # Print and list the files
 st.download_button(