vishwask commited on
Commit
496a0be
·
verified ·
1 Parent(s): c000673

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -11,6 +11,7 @@ from langchain.chains import ConversationChain
11
  from langchain.memory import ConversationBufferMemory
12
  from langchain.llms import HuggingFaceHub
13
  from langchain.memory import ConversationBufferWindowMemory
 
14
 
15
  from pathlib import Path
16
  import chromadb
@@ -79,7 +80,7 @@ def load_doc(list_file_path, chunk_size, chunk_overlap):
79
  # Processing for one document only
80
  # loader = PyPDFLoader(file_path)
81
  # pages = loader.load()
82
- loaders = [PyPDFLoader(x) for x in list_file_path]
83
  pages = []
84
  for loader in loaders:
85
  pages.extend(loader.load())
@@ -226,7 +227,7 @@ def demo():
226
  vector_db = gr.State()
227
  qa_chain = gr.State()
228
  collection_name = gr.State()
229
- pdf_directory = '/home/user/app/pdfs'
230
 
231
  def process_pdfs():
232
  # List all PDF files in the directory
 
11
  from langchain.memory import ConversationBufferMemory
12
  from langchain.llms import HuggingFaceHub
13
  from langchain.memory import ConversationBufferWindowMemory
14
+ from langchain_community.document_loaders import TextLoader
15
 
16
  from pathlib import Path
17
  import chromadb
 
80
  # Processing for one document only
81
  # loader = PyPDFLoader(file_path)
82
  # pages = loader.load()
83
+ loaders = [TextLoader(x) for x in list_file_path]
84
  pages = []
85
  for loader in loaders:
86
  pages.extend(loader.load())
 
227
  vector_db = gr.State()
228
  qa_chain = gr.State()
229
  collection_name = gr.State()
230
+ pdf_directory = '/home/user/app/texts/'
231
 
232
  def process_pdfs():
233
  # List all PDF files in the directory