url-summarizer / app.py
gsharma's picture
Update app.py
055db97
raw
history blame contribute delete
No virus
586 Bytes
import gradio as gr
from transformers import pipeline
import requests
def summarize_url(url):
r = requests.get(url)
url_contents = r.text
summarizer = pipeline("summarization",
model="facebook/bart-large-cnn",
max_length=1024, # or 512, or whatever your cut-off is
truncation=True
)
summarized_text = summarizer(url_contents, min_length=30, do_sample=False)
return summarized_text[0]['summary_text']
iface = gr.Interface(fn=summarize_url, inputs="text", outputs="text")
iface.launch()