Update app.py

app.py

@@ -9,8 +9,10 @@ import cv2
 import numpy as np
 from PIL import Image
 import torch
+from hf_hub_download import hf_hub_download
+
+
 
-# Adjust import paths as needed
 sys.path.append('Utils')
 sys.path.append('model')
 from model.CBAM.reunet_cbam import reunet_cbam
@@ -19,8 +21,8 @@ from model.unet import UNET
 from Utils.area import pixel_to_sqft, process_and_overlay_image
 from Utils.convert import read_pansharpened_rgb
 
-# Define base directory for Hugging Face Spaces
-BASE_DIR = "/user"
+
+BASE_DIR = hf_hub_download(repo_id="Pavan2k4/Building_area", repo_type="space")
 
 # Define subdirectories
 UPLOAD_DIR = os.path.join(BASE_DIR, "uploaded_images")