Spaces:

nikhildsst
/

RAG_Chatbot

Sleeping

RAG_Chatbot / embeddings /embedding_manager.py

Upload 6 files

1638245 verified almost 2 years ago

380 Bytes

# 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)

	# 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)