import gradio as gr
from transformers import pipeline

corrector = pipeline("text2text-generation", model="bmd1905/vietnamese-correction")

def add_diacritics(text):
    return corrector(text, max_length=128)[0]["generated_text"]

demo = gr.Interface(fn=add_diacritics, inputs="text", outputs="text", title="Vietnamese Diacritics Corrector")
demo.launch()