Update app.py

app.py

@@ -22,12 +22,25 @@ from model.unet import UNET
 from Utils.area import pixel_to_sqft, process_and_overlay_image
 from Utils.convert import read_pansharpened_rgb
 
+@st.cache_resource
+def load_model():
+    model = reunet_cbam()
+    model.load_state_dict(torch.load('latest.pth', map_location='cpu')['model_state_dict'])
+    model.eval()
+    return model
+
+@st.cache_resource
+def get_api():
+    hf_api = HfApi()
+    HF_TOKEN = st.secrets.get("HF_TOKEN")
+    if not HF_TOKEN:
+        st.error("HF_TOKEN not found")
+        st.stop()
+    return HF_TOKEN
+
+HF_TOKEN = get_api()
+
 
-hf_api = HfApi()
-HF_TOKEN = st.secrets.get("HF_TOKEN")
-if not HF_TOKEN:
-    st.error("HF_TOKEN not found.")
-    st.stop()
 
 REPO_ID = "Pavan2k4/Building_area"
 REPO_TYPE = "space"
@@ -63,12 +76,7 @@ for directory in [UPLOAD_DIR, MASK_DIR, PATCHES_DIR, PRED_PATCHES_DIR]:
     os.makedirs(directory, exist_ok=True)
 
 # Load model
-@st.cache_resource
-def load_model():
-    model = reunet_cbam()
-    model.load_state_dict(torch.load('latest.pth', map_location='cpu')['model_state_dict'])
-    model.eval()
-    return model
+
 
 model = load_model()