Update app.py

app.py

@@ -136,7 +136,13 @@ if uploaded_file is not None:
     
     image = np.log10(data+1)
     with colA:
+        st.markdown("""<style>[data-baseweb="select"] {margin-top: 6px;}</style>""", unsafe_allow_html=True)
         plot_image(image, scale)
+
+    with colB:
+        threshold = st.slider("", 0.0, 1.0, 0.4, 0.05, label_visibility="hidden")
+        y_pred = np.zeros((128,128))
+        plot_prediction(y_pred)
     
     if detect:
         data, wcs = cut(data, wcs, scale=scale)
@@ -149,14 +155,8 @@ if uploaded_file is not None:
             pred = np.rot90(pred, -j)
             y_pred += pred / 4
 
-        with colB:
-            threshold = st.slider("", 0.0, 1.0, 0.4, 0.05, label_visibility="hidden")
-
-            # Thresholding
-            y_pred = np.where(y_pred > threshold, y_pred, 0)
-            
-            plot_prediction(y_pred)
-
+        y_pred = np.where(y_pred > threshold, y_pred, 0
+        
         # if decompose:
         #     cavs = decompose_cavity(y_pred, )