from transformers import FlaxT5ForConditionalGeneration
import gradio as gr

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

def summarize(text):
    return f"{text} !"
    
iface = gr.Interface(fn=summarize, inputs="text", outputs="text")
iface.launch()