Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -58,7 +58,7 @@ class PDFChatbot:
|
|
| 58 |
if current_chunk:
|
| 59 |
chunks.append(" ".join(current_chunk))
|
| 60 |
|
| 61 |
-
db = FAISS.from_documents(
|
| 62 |
|
| 63 |
relevant_chunks = db.similarity_search(user_question, k=3)
|
| 64 |
relevant_chunks = [chunk.page_content for chunk in relevant_chunks]
|
|
|
|
| 58 |
if current_chunk:
|
| 59 |
chunks.append(" ".join(current_chunk))
|
| 60 |
|
| 61 |
+
db = FAISS.from_documents(chunks, HuggingFaceEmbeddings(model_name='bkai-foundation-models/vietnamese-bi-encoder'))
|
| 62 |
|
| 63 |
relevant_chunks = db.similarity_search(user_question, k=3)
|
| 64 |
relevant_chunks = [chunk.page_content for chunk in relevant_chunks]
|