space-demo / app.py
tawounfouet's picture
Create app.py
91c143d
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('"', '"')
return result[0]["summary_text"]