BHO commited on
Commit
905319c
·
1 Parent(s): ff3d51b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -3,6 +3,7 @@ import os
3
  from langchain.chains import RetrievalQA
4
  from langchain.llms import OpenAI
5
  from langchain.document_loaders import PyPDFLoader
 
6
  from langchain.text_splitter import CharacterTextSplitter
7
  from langchain.embeddings import OpenAIEmbeddings
8
  from langchain.vectorstores import Chroma
@@ -21,7 +22,8 @@ def qa_system(pdf_file, openai_key, prompt, chain_type, k):
21
  os.environ["OPENAI_API_KEY"] = openai_key
22
 
23
  # load document
24
- loader = PyPDFLoader(pdf_file.name)
 
25
  documents = loader.load()
26
 
27
  # split the documents into chunks
 
3
  from langchain.chains import RetrievalQA
4
  from langchain.llms import OpenAI
5
  from langchain.document_loaders import PyPDFLoader
6
+ from langchain.document_loaders import DirectoryLoader
7
  from langchain.text_splitter import CharacterTextSplitter
8
  from langchain.embeddings import OpenAIEmbeddings
9
  from langchain.vectorstores import Chroma
 
22
  os.environ["OPENAI_API_KEY"] = openai_key
23
 
24
  # load document
25
+ # loader = PyPDFLoader(pdf_file.name)
26
+ loader = DirectoryLoader(dir_path, glob="**/*.pdf", loader_cls=PDFLoader)
27
  documents = loader.load()
28
 
29
  # split the documents into chunks