Update app.py

app.py

@@ -1,7 +1,38 @@
 import gradio as gr
+from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
 
-def greet(name):
-    return "Hello " + name + "!!"
+def load_model(checkpoint_path):
+    # Load tokenizer và model từ Hugging Face Hub
+    tokenizer = AutoTokenizer.from_pretrained(checkpoint_path)
+    model = AutoModelForSeq2SeqLM.from_pretrained(checkpoint_path)
+    return model, tokenizer
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
-iface.launch()
+def correct_spelling(text):
+    # Thêm tiền tố nếu model yêu cầu (tuỳ vào cách model được huấn luyện)
+    input_text = text  # Thay đổi nếu cần
+    
+    # Tokenize và generate
+    inputs = tokenizer(input_text, return_tensors="pt", max_length=512, truncation=True)
+    outputs = model.generate(
+        **inputs,
+        max_length=512,
+        num_beams=5,
+        early_stopping=True
+    )
+    corrected_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
+    return corrected_text
+
+# Đường dẫn đến model của bạn trên Hugging Face Hub
+checkpoint_path = "Diezu/Batpho_v2"  # Thay bằng tên repository của bạn
+model, tokenizer = load_model(checkpoint_path)
+
+# Tạo giao diện Gradio
+demo = gr.Interface(
+    fn=correct_spelling,
+    inputs=gr.Textbox(placeholder="Nhập văn bản có lỗi chính tả..."),
+    outputs=gr.Textbox(label="Văn bản đã sửa"),
+    title="Demo Sửa Lỗi Chính Tả",
+    description="Sử dụng mô hình của bạn để sửa lỗi chính tả tiếng Việt."
+)
+
+demo.launch()