Spaces:
Sleeping
Sleeping
Update utils/DocsLoader.py
Browse files- utils/DocsLoader.py +2 -2
utils/DocsLoader.py
CHANGED
|
@@ -66,8 +66,8 @@ def load_and_chunk(url: str) -> list[Document]:
|
|
| 66 |
# sentence_docs.append(Document(page_content=sent, metadata=doc.metadata))
|
| 67 |
full_text = "\n".join([doc.page_content for doc in docs])
|
| 68 |
|
| 69 |
-
splitter = RecursiveCharacterTextSplitter(chunk_size=800, chunk_overlap=150)
|
| 70 |
-
|
| 71 |
# return splitter.split_documents(docs)
|
| 72 |
# return splitter.split_documents([Document(page_content=full_text)])
|
| 73 |
return splitter.create_documents([full_text])
|
|
|
|
| 66 |
# sentence_docs.append(Document(page_content=sent, metadata=doc.metadata))
|
| 67 |
full_text = "\n".join([doc.page_content for doc in docs])
|
| 68 |
|
| 69 |
+
# splitter = RecursiveCharacterTextSplitter(chunk_size=800, chunk_overlap=150)
|
| 70 |
+
splitter = SentenceTransformersTokenTextSplitter(model_name = MODEL_DIR,tokens_per_chunk=512, chunk_overlap=80)
|
| 71 |
# return splitter.split_documents(docs)
|
| 72 |
# return splitter.split_documents([Document(page_content=full_text)])
|
| 73 |
return splitter.create_documents([full_text])
|