from transformers import FlaxT5ForConditionalGeneration
import gradio as gr

model_flax = FlaxT5ForConditionalGeneration.from_pretrained("thegoodfellas/tgf-flan-t5-base-ptbr")

def summarize():
    pass

demo = gr.Interface(fn=summarize,
    inputs=[
        gr.inputs.Textbox(lines=20, label="Text"),
    ],
    outputs=gr.outputs.Textbox(label="Summary"),
    title="Flan T5 Summarization",
    description="Summarize text using T5 model",
)

demo.launch()