Update app.py

app.py

@@ -122,7 +122,6 @@ def decompose_cavity(pred, th2=0.7, amin=10):
         cavities.append(img)
 
     # Save raw and decomposed predictions to predictions folder
-    os.system("mkdir -p predictions")
     ccd = CCDData(pred, unit="adu", wcs=wcs)
     ccd.write(f"predictions/predicted.fits", overwrite=True)
     image_decomposed = np.zeros((128,128))
@@ -138,6 +137,8 @@ st.set_page_config(page_title="Cavity Detection Tool", layout="wide")
 bordersize = 0.6
 _, col, _ = st.columns([bordersize, 3, bordersize])
 
+os.system("mkdir -p predictions")
+
 with col:
     # Create heading and description
     st.markdown("# Cavity Detection Tool")