vishwask commited on
Commit
61b7a7c
·
1 Parent(s): 9d02a06

Update app.py

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