Spaces:
Runtime error
Runtime error
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 |
|