Update app.py

app.py

@@ -27,6 +27,7 @@ from werkzeug.utils import secure_filename
 from geopy.geocoders import Nominatim
 import pickle
 import numpy as np
+from huggingface_hub import snapshot_download
 
 # Configure logging
 logging.basicConfig(level=logging.INFO)
@@ -59,7 +60,14 @@ conversation_context = {}
 def load_sentence_transformer():
     print("Loading SentenceTransformer model...")
     try:
-        model_embedding = SentenceTransformer("jinaai/jina-embeddings-v3", trust_remote_code=True).to(device)
+        # First download the model
+        model_path = snapshot_download(
+            repo_id="jinaai/jina-embeddings-v3",
+            trust_remote_code=True
+        )
+        # Then load it with SentenceTransformer
+        model_embedding = SentenceTransformer(model_path).to(device)
+
         if not os.path.exists(MODEL_PATH):
             raise FileNotFoundError(f"Model state dict not found at {MODEL_PATH}. Please ensure the file exists.")
         state_dict = torch.load(MODEL_PATH, map_location=device)