singhankur01 commited on
Commit
76d0081
·
verified ·
1 Parent(s): b1a8e58

Update utils/DocsLoader.py

Browse files
Files changed (1) hide show
  1. 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
- # splitter = SentenceTransformersTokenTextSplitter(model_name = MODEL_DIR,tokens_per_chunk=500, 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])
 
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])