Update app.py

app.py

@@ -24,11 +24,13 @@ from huggingface_hub import HfApi, login
 import os
 
 # Set up Hugging Face authentication
+@st.cache_resource
 HF_TOKEN = os.environ.get("HF_TOKEN")
 if not HF_TOKEN:
     raise ValueError("HF_TOKEN environment variable is not set")
 
-login(token=HF_TOKEN)
+login(token=HF_TOKEN, add_to_git_credential=True)
+
 hf_api = HfApi()
 
 REPO_ID = "Pavan2k4/Building_area"
@@ -37,7 +39,7 @@ REPO_TYPE = "space"
 @st.cache_resource
 def load_model():
     model = reunet_cbam()
-    model.load_state_dict(torch.load('latest.pth', map_location='cpu')['model_state_dict'])
+    model.load_state_dict(torch.load('latest.pth', map_location='cpu', weights_only = True)['model_state_dict'])
     model.eval()
     return model
 
@@ -46,6 +48,7 @@ def load_model():
 
 
 
+
 # save to dir func
 def save_to_hf_repo(local_path, repo_path):
     try: