Update app.py

app.py

@@ -60,9 +60,9 @@ def plot_prediction(pred):
     with colB: st.pyplot()
 
 # Define function to plot the decomposed prediction
-def plot_decomposed(pred):
+def plot_decomposed(decomposed):
     plt.figure(figsize=(4, 4))
-    plt.imshow(pred, origin="lower")
+    plt.imshow(decomposed, origin="lower")
     plt.axis('off')
     with colC: st.pyplot()
         
@@ -172,9 +172,6 @@ if uploaded_file is not None:
         
     plot_prediction(y_pred)
 
-    # with colC:
-    #     st.markdown("""<style>[data-baseweb="select"] {margin-top: -36px;}</style>""", unsafe_allow_html=True)
-
     if decompose:
         y_pred = np.load("thresh.npy")
         
@@ -188,7 +185,7 @@ if uploaded_file is not None:
             ccd.write(f"predicted_{i+1}.fits", overwrite=True)
             image_decomposed += (i+1) * np.where(cav > 0, 1, 0)
 
-        shutil.make_archive("predictions.zip", 'zip', "predictions")
+        # shutil.make_archive("predictions.zip", 'zip', "predictions")
         np.save("decomposed.npy", image_decomposed)
 
     try: image_decomposed = np.load("decomposed.npy")