Update app.py

app.py

@@ -87,13 +87,12 @@ def save_to_hf_repo(local_path, repo_path):
         st.exception(e)
 
 
-BASE_DIR = os.getcwd()
+
 
 # Define subdirectories
-UPLOAD_DIR =  "uploaded_images"
-MASK_DIR =  "generated_masks"
-PATCHES_DIR = "patches"
-PRED_PATCHES_DIR =  "pred_patches"
+UPLOAD_DIR =  "Images"
+MASK_DIR =  "Masks"
+
 CSV_LOG_PATH =  "image_log.csv"
 
 # Create directories
@@ -229,8 +228,7 @@ def upload_page():
             filepath = os.path.join(UPLOAD_DIR, filename)
             converted_filepath = os.path.join(UPLOAD_DIR, converted_filename)
             
-            with open(filepath, "wb") as f:
-                f.write(bytes_data)
+            
 
             
             #st.success(f"Image saved to {filepath}")