Update app.py

app.py

@@ -130,6 +130,8 @@ def decompose_cavity(pred, th2=0.7, amin=10):
         ccd.write(f"predictions/predicted_{i+1}.fits", overwrite=True)
         image_decomposed += (i+1) * np.where(cav > 0, 1, 0)
 
+    shutil.make_archive("predictions", 'zip', "predictions")
+    
     return image_decomposed
 
 # Use wide layout and create columns