Update app.py

app.py

@@ -28,7 +28,7 @@ _, col, _ = st.columns([bordersize, 3, bordersize])
 with col:
     st.markdown("# Cavity Detection Tool")
     
-    st.markdown("Cavity Detection Tool (CADET) is a machine learning pipeline trained to detect X-ray cavities from noisy Chandra images of early-type galaxies.\ To use this tool: upload your image, select the scale of interest, and make a prediction!\ If you use this tool for your research, please cite [Plšek et al. 2023](https://arxiv.org/abs/2304.05457)")
+    st.markdown("Cavity Detection Tool (CADET) is a machine learning pipeline trained to detect X-ray cavities from noisy Chandra images of early-type galaxies.<br> To use this tool: upload your image, select the scale of interest, and make a prediction!<br> If you use this tool for your research, please cite [Plšek et al. 2023](https://arxiv.org/abs/2304.05457)")
     
     st.markdown("Input images should be centered at the centre of the galaxy and point sources should be filled with surrounding background ([dmfilth](https://cxc.cfa.harvard.edu/ciao/ahelp/dmfilth.html)).")
     
@@ -149,7 +149,11 @@ if uploaded_file is not None:
             pred = np.rot90(pred, -j)
             y_pred += pred / 4
 
+        np.save(y_pred, "pred.npy")
+
     if detect or bool(threshold):
+        y_pred = np.load("pred.npy")
+        
         pred = np.where(y_pred > threshold, y_pred, 0)
         
         plot_prediction(pred)