Create init_models.py

init_models.py

@@ -0,0 +1,20 @@
+import os
+from sentence_transformers import SentenceTransformer
+
+def preload_models():
+    """Pre-load all required models"""
+    try:
+        # Ensure cache directories exist with proper permissions
+        os.makedirs('/app/.cache/sentence_transformers', mode=0o777, exist_ok=True)
+        os.makedirs('/app/.cache/huggingface', mode=0o777, exist_ok=True)
+        
+        print("Loading sentence-transformers model...")
+        model = SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2')
+        print("Model loaded successfully!")
+        
+    except Exception as e:
+        print(f"Error loading models: {e}")
+        raise
+
+if __name__ == "__main__":
+    preload_models()