Spaces:
Sleeping
Sleeping
File size: 561 Bytes
4e7e4c0 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 | from sentence_transformers import SentenceTransformer
import numpy as np
from typing import List
class EmbeddingModel:
def __init__(self , model_name:str = "paraphrase-multilingual-MiniLM-L12-v2"):
self.model_name = model_name
self.model = SentenceTransformer(
model_name)
def encode(self , texts: List[str])-> np.ndarray:
return self.model.encode(
texts,
show_progress_bar=False,
convert_to_numpy=True,
normalize_embeddings=True
)
|