Spaces:
Sleeping
Sleeping
File size: 380 Bytes
1638245 | 1 2 3 4 5 6 7 8 9 10 11 12 13 | # embeddings/embedding_manager.py
from sentence_transformers import SentenceTransformer
from config import Config
class EmbeddingManager:
def __init__(self):
self.model = SentenceTransformer(Config.EMBEDDING_MODEL)
def get_embedding(self, text):
return self.model.encode(text)
def get_embeddings(self, texts):
return self.model.encode(texts) |