init_models.py

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()