singhankur01 commited on
Commit
c739107
·
verified ·
1 Parent(s): 1706a16

Update utils/DocsLoader.py

Browse files
Files changed (1) hide show
  1. 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=600, chunk_overlap=80,separators=["\n\n", "\n", ".", " ", ""])
54
- # splitter = SentenceTransformersTokenTextSplitter(model_name = MODEL_DIR,tokens_per_chunk=500, chunk_overlap=80)
55
  return splitter.split_documents(docs)
 
50
  else:
51
  raise HTTPException(400, f"Unsupported document type: {content_type}")
52
 
53
+ # splitter = RecursiveCharacterTextSplitter(chunk_size=600, chunk_overlap=80,separators=["\n\n", "\n", ".", " ", ""])
54
+ splitter = SentenceTransformersTokenTextSplitter(model_name = MODEL_DIR,tokens_per_chunk=500, chunk_overlap=80)
55
  return splitter.split_documents(docs)