Spaces:
Sleeping
Sleeping
File size: 204 Bytes
8972ad7 |
1 2 3 4 5 6 7 |
import numpy as np
from sentence_transformers import SentenceTransformer
def extract_embedding(text: str, model: SentenceTransformer) -> np.ndarray:
return model.encode(text, convert_to_numpy=True)
|