quoc-khanh commited on
Commit
4f8fe62
·
verified ·
1 Parent(s): 7ecbb5e

Update file_loader.py

Browse files
Files changed (1) hide show
  1. file_loader.py +2 -1
file_loader.py CHANGED
@@ -51,7 +51,8 @@ def get_vectorstore():
51
  # Lưu vào vectorstore với nhúng từ Google GenAI
52
  # embedding = GoogleGenerativeAIEmbeddings(model="models/text-embedding-004")
53
  print('Get embedding model /paraphrase-multilingual-MiniLM-L12-v2')
54
- embedding = HuggingFaceEmbeddings(model_name="VoVanPhuc/sup-SimCSE-VietNamese-phobert-base")
55
  print('Set vectorstore FAISS')
56
  vectorstore = FAISS.from_documents(documents=all_splits, embedding=embedding)
 
57
  return vectorstore
 
51
  # Lưu vào vectorstore với nhúng từ Google GenAI
52
  # embedding = GoogleGenerativeAIEmbeddings(model="models/text-embedding-004")
53
  print('Get embedding model /paraphrase-multilingual-MiniLM-L12-v2')
54
+ embedding = HuggingFaceEmbeddings(model_name="sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2")
55
  print('Set vectorstore FAISS')
56
  vectorstore = FAISS.from_documents(documents=all_splits, embedding=embedding)
57
+ print('Vectorstore ready!')
58
  return vectorstore