deploy

models/model_loader.py

@@ -1,4 +1,15 @@
 from sentence_transformers import SentenceTransformer
+import os
 
-# Load the pre-trained model once at startup
-model = SentenceTransformer('Alibaba-NLP/gte-base-en-v1.5', trust_remote_code=True)     
+# Model path and Hugging Face model ID
+model_dir = 'models/sentence_transformer'
+hf_model_id = 'Alibaba-NLP/gte-base-en-v1.5'
+
+# Check if model directory exists
+if not os.path.exists(model_dir):
+    print(f"Downloading model '{hf_model_id}' from Hugging Face...")
+    model = SentenceTransformer(hf_model_id, trust_remote_code=True)
+    model.save(model_dir)
+else:
+    print(f"Loading model from local directory: {model_dir}")
+    model = SentenceTransformer(model_dir)