Update app.py

app.py

@@ -48,7 +48,8 @@ if __name__ == "__main__":
     st.title("Sartorius Cell Segmentation")
 
     uploaded_img = st.file_uploader(label="Upload neuronal cell image")
-    if uploaded_img is not None:
+    segment = st.button("Perform segmentation")
+    if uploaded_img is not None and segment:
         img = Image.open(uploaded_img)
         img = np.array(img)
         st.image(img)
@@ -65,4 +66,4 @@ if __name__ == "__main__":
             **model_params
         )
 
-        print(preds)
+        print(preds)