vishwask commited on
Commit
0880fcf
·
1 Parent(s): a9a0abe

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -53,7 +53,7 @@ st.markdown(f"""<style>
53
 
54
  DEVICE = "cuda:0" if torch.cuda.is_available() else "cpu"
55
 
56
- loader = PyPDFDirectoryLoader("/pdfs/")
57
  docs = loader.load()
58
  #len(docs)
59
 
@@ -222,9 +222,9 @@ if prompt := st.chat_input("How can I help you today?"):
222
  pdf_document.close()
223
 
224
 
225
- pdf_page_to_image('/pdf2image/output.pdf', page_number, '/pdf2image/output.png')
226
 
227
- image = Image.open('/pdf2image/output.png')
228
  st.image(image)
229
  st.audio(sound_file)
230
 
 
53
 
54
  DEVICE = "cuda:0" if torch.cuda.is_available() else "cpu"
55
 
56
+ loader = PyPDFDirectoryLoader("./pdfs/")
57
  docs = loader.load()
58
  #len(docs)
59
 
 
222
  pdf_document.close()
223
 
224
 
225
+ pdf_page_to_image('./pdf2image/output.pdf', page_number, './pdf2image/output.png')
226
 
227
+ image = Image.open('./pdf2image/output.png')
228
  st.image(image)
229
  st.audio(sound_file)
230