Spaces:
Sleeping
Sleeping
Update utils/DocsLoader.py
Browse files- utils/DocsLoader.py +2 -2
utils/DocsLoader.py
CHANGED
|
@@ -50,6 +50,6 @@ def load_and_chunk(url: str) -> list[Document]:
|
|
| 50 |
else:
|
| 51 |
raise HTTPException(400, f"Unsupported document type: {content_type}")
|
| 52 |
|
| 53 |
-
splitter = RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=120,separators=["\n\n", "\n", ".", " ", ""])
|
| 54 |
-
|
| 55 |
return splitter.split_documents(docs)
|
|
|
|
| 50 |
else:
|
| 51 |
raise HTTPException(400, f"Unsupported document type: {content_type}")
|
| 52 |
|
| 53 |
+
# splitter = RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=120,separators=["\n\n", "\n", ".", " ", ""])
|
| 54 |
+
splitter = SentenceTransformersTokenTextSplitter(model_name = MODEL_DIR,tokens_per_chunk=500, chunk_overlap=80)
|
| 55 |
return splitter.split_documents(docs)
|