File size: 381 Bytes
f499d4b
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
from sentence_transformers import SentenceTransformer
import sys

try:
    model = SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2', device='cpu')
    print("Model loaded successfully.")
    embeddings = model.encode(["Hello world"])
    print(f"Embeddings shape: {embeddings.shape}")
except Exception as e:
    print(f"Error loading/using model: {e}")
    sys.exit(1)