chukbert commited on
Commit
24327f4
·
verified ·
1 Parent(s): 115d673

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -11,11 +11,12 @@ from langchain.prompts import PromptTemplate
11
  from langchain_community.document_loaders import PyPDFLoader
12
  from langchain_openai import ChatOpenAI
13
  from langchain.callbacks import get_openai_callback
 
14
 
15
  openai.api_key = os.getenv("OPENAI_API_KEY")
16
 
17
  def process_pdf(file):
18
- loader = PyPDFLoader(file)
19
  documents = loader.load()
20
  text_splitter = RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=200)
21
  splits = text_splitter.split_documents(documents)
 
11
  from langchain_community.document_loaders import PyPDFLoader
12
  from langchain_openai import ChatOpenAI
13
  from langchain.callbacks import get_openai_callback
14
+ from io import BytesIO
15
 
16
  openai.api_key = os.getenv("OPENAI_API_KEY")
17
 
18
  def process_pdf(file):
19
+ loader = PyPDFLoader(BytesIO(file.read()))
20
  documents = loader.load()
21
  text_splitter = RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=200)
22
  splits = text_splitter.split_documents(documents)