space-demo / app.py
JLuissp's picture
Update app.py
e91ce67
raw
history blame
658 Bytes
from transformers import pipeline
import gradio as gr
summarizer = pipeline('summarization',
model='t5-base',
tokenizer='t5-small',
truncation=True,
framework='tf')
def translate(text):
text = text.replace('"', '"')
text = text.replace(''', "'")
text = text.replace('&', "&")
result =summarizer(text, min_lenght=180, truncation=True)
return result[0]['summary_text']
iface = gr.Interface(
fn = translate,
inputs = gr.inputs.Textbox(lines=10, placeholder='Enter text to summarize...'),
outputs='Text'
)
iface.launch()