tony346 commited on
Commit
4ba6eb8
1 Parent(s): 4e31709

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -5
app.py CHANGED
@@ -8,15 +8,16 @@ from langchain.chains import ConversationalRetrievalChain
8
  from htmlTemplates import css, bot_template, user_template
9
  from langchain.llms import LlamaCpp # For loading transformer models.
10
  from langchain.document_loaders import PyPDFLoader, TextLoader, JSONLoader, CSVLoader
11
- from tempfile import NamedTemporaryFile
12
  from huggingface_hub import hf_hub_download
13
 
14
 
15
  def get_pdf_text(pdf_docs):
16
- # with NamedTemporaryFile() as temp_file:
17
- # temp_file.write(pdf_docs.getvalue())
18
- # temp_file.seek(0)
19
- pdf_loader = PyPDFLoader(pdf_docs)
 
20
  pdf_doc = pdf_loader.load()
21
  return pdf_doc
22
 
 
8
  from htmlTemplates import css, bot_template, user_template
9
  from langchain.llms import LlamaCpp # For loading transformer models.
10
  from langchain.document_loaders import PyPDFLoader, TextLoader, JSONLoader, CSVLoader
11
+ import tempfile
12
  from huggingface_hub import hf_hub_download
13
 
14
 
15
  def get_pdf_text(pdf_docs):
16
+ temp_dir = tempfile.TemporaryDirectory()
17
+ temp_filepath = os.path.join(temp_dir.name, pdf_docs.name)
18
+ with open(temp_filepath, "wb") as f:
19
+ f.write(pdf_docs.getvalue())
20
+ pdf_loader = PyPDFLoader(temp_filepath)
21
  pdf_doc = pdf_loader.load()
22
  return pdf_doc
23