Update app.py

app.py

@@ -18,7 +18,7 @@ from model.CBAM.reunet_cbam import reunet_cbam
 from model.transform import transforms
 from model.unet import UNET
 from Utils.area import pixel_to_sqft, process_and_overlay_image
-from split_merge import split, merge 
+
 from Utils.convert import read_pansharpened_rgb
 
 # Initialize Hugging Face API
@@ -40,7 +40,9 @@ MASK_DIR = "generated_masks"
 PATCHES_DIR = "patches"
 PRED_PATCHES_DIR = "pred_patches"
 CSV_LOG_PATH = "image_log.csv"
-
+# Create directories
+for directory in [UPLOAD_DIR, MASK_DIR, PATCHES_DIR, PRED_PATCHES_DIR]:
+    os.makedirs(directory, exist_ok=True)
 
 def split(image, destination = PATCHES_DIR, patch_size = 650):
     img = cv2.imread(image)
@@ -86,9 +88,7 @@ def merge(patch_folder , dest_image = 'out.png', image_shape = None):
 
     cv2.imwrite(dest_image, merged)
 
-# Create directories
-for directory in [UPLOAD_DIR, MASK_DIR, PATCHES_DIR, PRED_PATCHES_DIR]:
-    os.makedirs(directory, exist_ok=True)
+
 
 # Load model
 @st.cache_resource