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}")