Spaces:
Runtime error
Runtime error
File size: 549 Bytes
79eca0e 84cf206 79eca0e 84cf206 ea5aa8c 84cf206 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
from sentence_transformers import SentenceTransformer
import os
# Model path and Hugging Face model ID
model_dir = '/tmp/sentence_transformer' # Change here
hf_model_id = 'Alibaba-NLP/gte-base-en-v1.5'
# Check if model directory exists
if not os.path.exists(model_dir):
print(f"Downloading model '{hf_model_id}' from Hugging Face...")
model = SentenceTransformer(hf_model_id, trust_remote_code=True)
model.save(model_dir)
else:
print(f"Loading model from local directory: {model_dir}")
model = SentenceTransformer(model_dir)
|