Update app.py

app.py

@@ -34,19 +34,19 @@ if not HF_TOKEN:
 REPO_ID = "Pavan2k4/Building_area"
 REPO_TYPE = "space"
 
-# Define subdirectories using relative paths
-UPLOAD_DIR = "uploaded_images"
-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]:
+# Define temporary directories outside of tracked directories
+TMP_DIR = "/tmp/space_files"
+PATCHES_DIR = os.path.join(TMP_DIR, "patches")
+PRED_PATCHES_DIR = os.path.join(TMP_DIR, "pred_patches")
+MASK_DIR = os.path.join(TMP_DIR, "generated_masks")
+
+# Create temporary directories
+os.makedirs(TMP_DIR, exist_ok=True)
+for directory in [PATCHES_DIR, PRED_PATCHES_DIR, MASK_DIR]:
     os.makedirs(directory, exist_ok=True)
 
-# Load model
-@st.cache_resource
+# Load model and cache it to avoid unnecessary reloading
+@st.cache_resource(show_spinner=False)
 def load_model():
     model = reunet_cbam()
     model.load_state_dict(torch.load('latest.pth', map_location='cpu')['model_state_dict'])
@@ -111,7 +111,8 @@ def split(image_path, patch_size=512):
             patch_filename = f"patch_{i}_{j}.png"
             patch_path = os.path.join(PATCHES_DIR, patch_filename)
             patch.save(patch_path)
-            st.write(f"Saved patch: {patch_path}")  # Debug output
+            st.write(f"Saved patch: {patch_path}")
+
 
 def upload_page():
     if 'file_uploaded' not in st.session_state: