Spaces:
Runtime error
Runtime error
| 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() |