Text_Summarizer / app.py
AyushDey's picture
Changed model to bart-large-cnn
fbcbcc2
raw
history blame
302 Bytes
import gradio as gr
from transformers import pipeline
pipe = pipeline('summarization', model='facebook/bart-large-cnn')
def summary(input):
value = pipe(input)
return value[0]['summary_text']
a = gr.Interface(fn=summary, inputs= 'text', outputs = 'text', title = 'Text Summarizer')
a.launch()