tensorboy0101 commited on
Commit
8cd7e1e
·
verified ·
1 Parent(s): 7a1a7e1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,5 +1,5 @@
1
  from dotenv import load_dotenv
2
-
3
  load_dotenv()
4
  import base64
5
  import streamlit as st
@@ -19,7 +19,7 @@ def get_gemini_response(input,pdf_cotent,prompt):
19
  def input_pdf_setup(uploaded_file):
20
  if uploaded_file is not None:
21
  ## Convert the PDF to image
22
- images=pdf2image.convert_from_bytes(uploaded_file.read())
23
 
24
  first_page=images[0]
25
 
 
1
  from dotenv import load_dotenv
2
+ import fitz
3
  load_dotenv()
4
  import base64
5
  import streamlit as st
 
19
  def input_pdf_setup(uploaded_file):
20
  if uploaded_file is not None:
21
  ## Convert the PDF to image
22
+ images=fitz.open(stream=pdf_bytes, filetype="pdf")
23
 
24
  first_page=images[0]
25