Update app.py

app.py

@@ -22,7 +22,7 @@ st.set_page_config(page_title="Cavity Detection Tool", layout="wide")
 
 # st.title("Cavity Detection Tool")
 
-bordersize = 0.8
+bordersize = 0.6
 _, col, _ = st.columns([bordersize, 3, bordersize])
 
 with col:
@@ -151,7 +151,7 @@ if uploaded_file is not None:
         # Thresholding
         y_pred = np.where(y_pred > threshold, y_pred, 0)
                 
-        plot_prediction(y_pred, decompose)
+        plot_prediction(y_pred)
 
         # if decompose:
         #     cavs = decompose_cavity(y_pred, )