Update app.py

app.py

@@ -152,7 +152,7 @@ if uploaded_file is not None:
         data = hdul[0].data
         wcs = WCS(hdul[0].header)
     
-    # Make four columns for buttons
+    # Make six columns for buttons
     _, col1, col2, col3, col4, col5, col6, _ = st.columns([bordersize,0.5,0.5,0.5,0.5,0.5,0.5,bordersize])
     col1.subheader("Input image")
     col3.subheader("Prediction")
@@ -182,7 +182,7 @@ if uploaded_file is not None:
             threshold = st.slider("", 0.0, 1.0, 0.0, 0.05, label_visibility="hidden")
     
     if detect:
-        y_pred = predict(data, scale)
+        y_pred = cut_n_predict(data, scale)
         
         # np.save("pred.npy", y_pred)