Spaces:
Sleeping
Sleeping
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)
|