quoc-khanh commited on
Commit
bb31a11
·
verified ·
1 Parent(s): b62e2fc

Update file_loader.py

Browse files
Files changed (1) hide show
  1. file_loader.py +1 -1
file_loader.py CHANGED
@@ -75,7 +75,7 @@ def get_vectorstore():
75
  # Lưu vào vectorstore với nhúng từ Google GenAI
76
  # embedding = GoogleGenerativeAIEmbeddings(model="models/text-embedding-004")
77
  print('Get embedding model /paraphrase-multilingual-MiniLM-L12-v2')
78
- embedding = HuggingFaceEmbeddings(model_name="VoVanPhuc/sup-SimCSE-VietNamese-phobert-base")
79
  print('Set vectorstore FAISS')
80
  vectorstore = FAISS.from_documents(documents=all_splits, embedding=embedding)
81
  return vectorstore
 
75
  # Lưu vào vectorstore với nhúng từ Google GenAI
76
  # embedding = GoogleGenerativeAIEmbeddings(model="models/text-embedding-004")
77
  print('Get embedding model /paraphrase-multilingual-MiniLM-L12-v2')
78
+ embedding = HuggingFaceEmbeddings(model_name="sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2") #"VoVanPhuc/sup-SimCSE-VietNamese-phobert-base")
79
  print('Set vectorstore FAISS')
80
  vectorstore = FAISS.from_documents(documents=all_splits, embedding=embedding)
81
  return vectorstore