space-demo / app.py
andrea777h's picture
Update app.py
db87278
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_length=180,truncation=True,framework="tf")
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()