Update app.py

app.py

@@ -180,7 +180,7 @@ def upload_page():
 
                 # Count and display the number of patches
                 num_patches = len([f for f in os.listdir(PATCHES_DIR) if f.endswith('.png')])
-                st.write(f"Number of patches created: {num_patches}")  # Debug output
+                #st.write(f"Number of patches created: {num_patches}")  # Debug output
 
                 # Display buffer while analyzing
                 with st.spinner('Analyzing...'):
@@ -188,7 +188,7 @@ def upload_page():
                     for patch_filename in os.listdir(PATCHES_DIR):
                         if patch_filename.endswith(".png"):
                             patch_path = os.path.join(PATCHES_DIR, patch_filename)
-                            st.write(f"Processing patch: {patch_path}")  # Debug output
+                            #st.write(f"Processing patch: {patch_path}")  # Debug output
                             patch_img = Image.open(patch_path)
                             patch_tr_img = transforms(patch_img)
                             prediction = predict(patch_tr_img)
@@ -196,7 +196,7 @@ def upload_page():
                             mask_filename = f"mask_{patch_filename}"
                             mask_filepath = os.path.join(PRED_PATCHES_DIR, mask_filename)
                             Image.fromarray(mask).save(mask_filepath)
-                            st.write(f"Saved mask: {mask_filepath}")  # Debug output
+                            #st.write(f"Saved mask: {mask_filepath}")  # Debug output
 
                     # Merge predicted patches
                     merged_mask_filename = f"mask_{timestamp}.png"