import gradio as gr
import transformers
import torch
from transformers import pipeline

summarizer = pipeline("summarization", model="google-t5/t5-base", tokenizer="google-t5/t5-base", framework="pt")

def summarize(text):
    return summarizer(text, max_length=20)
    

demo = gr.Interface(fn=summarize, inputs="text", outputs="text")
demo.launch()