ddovidovich commited on
Commit
18baae6
·
1 Parent(s): 22b0051

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -36,9 +36,10 @@ if uploaded_file:
36
  img.save(file_name+'.pdf')
37
  PDFFileName = file_name+'.pdf'
38
  else:
39
- with NamedTemporaryFile(dir='.', suffix='.pdf') as f:
40
  f.write(uploaded_file.getbuffer())
41
  PDFFileName = f.name
 
42
  loader = UnstructuredPDFLoader(PDFFileName)
43
  pages = loader.load_and_split()
44
  embeddings = OpenAIEmbeddings()
@@ -50,3 +51,4 @@ if uploaded_file:
50
  chain = load_qa_chain(ChatOpenAI(temperature=0), chain_type="stuff")
51
  output = chain.run(input_documents=docs, question=query)
52
  st.write(output)
 
 
36
  img.save(file_name+'.pdf')
37
  PDFFileName = file_name+'.pdf'
38
  else:
39
+ with NamedTemporaryFile(delete=False, dir='.', suffix='.pdf') as f:
40
  f.write(uploaded_file.getbuffer())
41
  PDFFileName = f.name
42
+ st.write("Document parsing in progress ...")
43
  loader = UnstructuredPDFLoader(PDFFileName)
44
  pages = loader.load_and_split()
45
  embeddings = OpenAIEmbeddings()
 
51
  chain = load_qa_chain(ChatOpenAI(temperature=0), chain_type="stuff")
52
  output = chain.run(input_documents=docs, question=query)
53
  st.write(output)
54
+ st.write("Done...")