Update app.py

app.py

@@ -2,6 +2,14 @@ import streamlit as st
 from transformers import pipeline
 from pathlib import Path
 from llama_index import download_loader
+from PyPDF2 import PdfReader
+
+def get_pdf_text(pdf_docs):
+    text = ""
+    pdf_reader = PdfReader(pdf)
+    for page in pdf_reader.pages:
+        text += page.extract_text()
+    return text
 
 def main():
   st.title("PDF Summarizer")
@@ -13,9 +21,10 @@ def main():
     # data Loader for reading PDF
     PDFReader = download_loader("PDFReader")
     loader = PDFReader()
-    documents = loader.load_data(file=Path(uploaded_file))
-    print(type(documents))
-    summary = summarizer(str(documents[0]), max_length=50, min_length=30, do_sample=False)
+    #documents = loader.load_data(file=Path(uploaded_file))
+    #print(type(documents))
+    text  = get_pdf_text(uploaded_file)
+    summary = summarizer(text, max_length=50, min_length=30, do_sample=False)
     st.write(summary)
 
 if __name__ == "__main__":