Update app.py

app.py

@@ -4,7 +4,7 @@ model = from_pretrained_keras("Plsek/CADET-v1")
 
 # Basic libraries
 import os
-import shutil
+# import shutil
 import numpy as np
 from scipy.ndimage import center_of_mass
 import matplotlib.pyplot as plt
@@ -164,7 +164,7 @@ if uploaded_file is not None:
         st.markdown("""<style>[data-baseweb="select"] {margin-top: -36px;}</style>""", unsafe_allow_html=True)
         threshold = st.slider("", 0.0, 1.0, 0.0, 0.05, label_visibility="hidden")
     
-    if detect:
+    if detect or threshold:
         data, wcs = cut(data, wcs, scale=scale)
         image = np.log10(data+1)
         
@@ -175,12 +175,12 @@ if uploaded_file is not None:
             pred = np.rot90(pred, -j)
             y_pred += pred / 4
 
-        np.save("pred.npy", y_pred)
+        # np.save("pred.npy", y_pred)
 
-    try: y_pred = np.load("pred.npy")
-    except: y_pred = np.zeros((128,128))
-    # try: y_pred
+    # try: y_pred = np.load("pred.npy")
     # except: y_pred = np.zeros((128,128))
+    try: y_pred
+    except: y_pred = np.zeros((128,128))
     y_pred_th = np.where(y_pred > threshold, y_pred, 0)
     # np.save("thresh.npy", y_pred)