Samizie commited on
Commit
871e14b
·
verified ·
1 Parent(s): dc041ba

Update embedding/vector_store.py

Browse files
Files changed (1) hide show
  1. embedding/vector_store.py +4 -4
embedding/vector_store.py CHANGED
@@ -3,12 +3,12 @@ import shutil
3
  from langchain_huggingface import HuggingFaceEmbeddings
4
  from langchain_community.vectorstores import Chroma
5
 
6
- #embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2")
7
 
8
- from sentence_transformers import SentenceTransformer
9
 
10
- model = SentenceTransformer("all-MiniLM-L6-v2", trust_remote_code=True)
11
- embeddings = HuggingFaceEmbeddings(model_name=model_name)
12
 
13
  #model = SentenceTransformer("sentence-transformers/all-MiniLM-L6-v2", trust_remote_code=True)
14
 
 
3
  from langchain_huggingface import HuggingFaceEmbeddings
4
  from langchain_community.vectorstores import Chroma
5
 
6
+ embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2")
7
 
8
+ #from sentence_transformers import SentenceTransformer
9
 
10
+ #model = SentenceTransformer("all-MiniLM-L6-v2", trust_remote_code=True)
11
+ #embeddings = HuggingFaceEmbeddings(model_name=model_name)
12
 
13
  #model = SentenceTransformer("sentence-transformers/all-MiniLM-L6-v2", trust_remote_code=True)
14