Update models new

app.py

@@ -1,14 +1,10 @@
-# app.py for Gradio
 import gradio as gr
-from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+from transformers import pipeline
 
-tokenizer = AutoTokenizer.from_pretrained("bmd1905/vietnamese-correction-v2")
-model = AutoModelForSeq2SeqLM.from_pretrained("bmd1905/vietnamese-correction-v2")
+corrector = pipeline("text2text-generation", model="bmd1905/vietnamese-correction")
 
 def add_diacritics(text):
-    inputs = tokenizer(text, return_tensors="pt")
-    outputs = model.generate(**inputs, max_new_tokens=256)
-    return tokenizer.decode(outputs[0], skip_special_tokens=True)
+    return corrector(text, max_length=128)[0]["generated_text"]
 
-demo = gr.Interface(fn=add_diacritics, inputs="text", outputs="text", title="Vietnamese Diacritics Restorer")
+demo = gr.Interface(fn=add_diacritics, inputs="text", outputs="text", title="Vietnamese Diacritics Corrector")
 demo.launch()