dhiraj5678's picture
Update app.py
c680aaf
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
from io import StringIO
tokenizer = AutoTokenizer.from_pretrained("facebook/bart-large-cnn")
model = AutoModelForSeq2SeqLM.from_pretrained("facebook/bart-large-cnn")
import streamlit as st
import os
import PyPDF2
import fitz
uploaded_pdf = st.file_uploader("Load pdf: ", type=['pdf'])
if uploaded_pdf is not None:
doc = fitz.open(stream=uploaded_pdf.read(), filetype="pdf")
text = ""
for page in doc:
text += page.getText()
st.write(text)
doc.close()