Spaces:
Sleeping
Sleeping
File size: 1,126 Bytes
ea9ca44 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 |
try:
print("π¦ Imports check...")
import sentence_transformers
from sentence_transformers import SentenceTransformer
import torch
print(f"β
sentence-transformers version: {sentence_transformers.__version__}")
print(f"β
torch version: {torch.__version__}")
print("\nπ Attempting to load BGE-M3 model (this might trigger download)...")
# Using 'cpu' to allow it to run on any environment for this test
# but the actual code uses cuda if available
model = SentenceTransformer('BAAI/bge-m3', device='cpu')
print("β
Model loaded successfully!")
test_text = "This is a test resume sentence."
embedding = model.encode(test_text)
print(f"β
Generated embedding shape: {embedding.shape}")
if embedding.shape[0] == 1024:
print("β
SUCCESS: Embedding dimension is 1024.")
else:
print(f"β ERROR: Expected 1024 dimensions, got {embedding.shape[0]}")
except ImportError as e:
print(f"β Missing Dependency: {e}")
print("Run: pip install -r requirements.txt")
except Exception as e:
print(f"β Error: {e}")
|