Update app.py
Browse files
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 |
-
|
12 |
from huggingface_hub import hf_hub_download
|
13 |
|
14 |
|
15 |
def get_pdf_text(pdf_docs):
|
16 |
-
|
17 |
-
|
18 |
-
|
19 |
-
|
|
|
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 |
|