Spaces:
Sleeping
Sleeping
Update file_loader.py
Browse files- file_loader.py +1 -1
file_loader.py
CHANGED
|
@@ -51,7 +51,7 @@ 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="
|
| 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="VoVanPhuc/sup-SimCSE-VietNamese-phobert-base")
|
| 55 |
print('Set vectorstore FAISS')
|
| 56 |
vectorstore = FAISS.from_documents(documents=all_splits, embedding=embedding)
|
| 57 |
return vectorstore
|