Gunasai1907's picture
Create app.py
d8c9f63 verified
raw
history blame contribute delete
728 Bytes
from transformers import pipeline
import gradio as gr
import PyPDF2
summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
def text_summarize(pdf_file):
# Read PDF
text = ""
reader = PyPDF2.PdfReader(pdf_file.name)
for page in reader.pages:
text += page.extract_text()
# summarization logic
article = text[:1500] #(just demo: first 1500 chars)
response = summarizer(article)
return response[0]['summary_text']
# creating the user interface using gradio
demo = gr.Interface(
fn=text_summarize,
inputs=gr.File(type="filepath", file_types=[".pdf"]),
outputs="text",
title="Text Summarizer",
description="Upload a PDF to summarize"
)
demo.launch()