AptlyDigital commited on
Commit
901d782
·
verified ·
1 Parent(s): 451a637

Upload 3 files

Browse files
models/__init__.py ADDED
@@ -0,0 +1 @@
 
 
1
+ # Empty init files
models/embedder.py ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from sentence_transformers import SentenceTransformer
2
+ import chromadb
3
+
4
+ class Embedder:
5
+ def __init__(self, model_name="all-MiniLM-L6-v2"):
6
+ self.model = SentenceTransformer(model_name)
7
+ self.client = chromadb.Client()
8
+ self.collection = self.client.create_collection(name="documents")
9
+
10
+ def embed(self, texts):
11
+ return self.model.encode(texts).tolist()
models/llm_handler.py ADDED
@@ -0,0 +1 @@
 
 
1
+