Update app.py
Browse files
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("
|
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('
|
228 |
|
229 |
-
image = Image.open('
|
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 |
|