medelharchaoui commited on
Commit
fd18c44
1 Parent(s): 1e00a0f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -3
app.py CHANGED
@@ -1,4 +1,5 @@
1
  import streamlit as st
 
2
  import torch
3
  from langchain.document_loaders import PyPDFLoader
4
 
@@ -9,9 +10,11 @@ from langchain.vectorstores import FAISS
9
 
10
 
11
  def read_pdf(file):
12
- loader = PyPDFLoader(file)
13
- raw_documents = loader.load()
14
- return raw_documents
 
 
15
 
16
  st.title('PDF Text Extractor')
17
 
 
1
  import streamlit as st
2
+ import tempfile
3
  import torch
4
  from langchain.document_loaders import PyPDFLoader
5
 
 
10
 
11
 
12
  def read_pdf(file):
13
+ with tempfile.NamedTemporaryFile(delete=False) as temp:
14
+ temp.write(file.getvalue())
15
+ loader = PyPDFLoader(temp.name)
16
+ raw_documents = loader.load()
17
+ return raw_documents
18
 
19
  st.title('PDF Text Extractor')
20