download_model.py•1.15 kB
#!/usr/bin/env python3
"""
Download and verify the sentence-transformer model
"""
import sys
from pathlib import Path
def download_model():
"""Download the model if not already present"""
try:
print("🤖 Checking AI model...")
from sentence_transformers import SentenceTransformer
model_name = 'all-MiniLM-L6-v2'
print(f" 📥 Loading model: {model_name}")
print(" (This will download ~90MB on first run)")
# This will download if not cached
model = SentenceTransformer(model_name)
# Test the model
test_embedding = model.encode("test", show_progress_bar=False)
print(f" ✅ Model ready! Embedding dimension: {len(test_embedding)}")
return True
except ImportError:
print(" ❌ sentence-transformers not installed")
print(" Run: pip install sentence-transformers")
return False
except Exception as e:
print(f" ❌ Failed to load model: {e}")
return False
if __name__ == "__main__":
success = download_model()
sys.exit(0 if success else 1)