Update app.py

app.py

@@ -1,35 +1,40 @@
 import gradio as gr
-from transformers import pipeline
+from transformers import MarianMTModel, MarianTokenizer
 
-# Khởi tạo pipeline dịch thuật với mô hình Helsinki-NLP
-translator = pipeline("translation_zh_to_vi", model="Helsinki-NLP/opus-mt-zh-vi")
+MODEL_NAME = "Helsinki-NLP/opus-mt-zh-vi"
 
-def translate_zh_to_vi(text):
-    # Xử lý trường hợp đầu vào rỗng
-    if not text or text.strip() == "":
+print("Đang tải model...")
+tokenizer = MarianTokenizer.from_pretrained(MODEL_NAME)
+model = MarianMTModel.from_pretrained(MODEL_NAME)
+print("Tải model thành công!")
+
+def translate(text: str) -> str:
+    if not text.strip():
         return ""
-    
-    # Thực hiện dịch
-    result = translator(text)
-    return result[0]['translation_text']
+    inputs = tokenizer(text, return_tensors="pt", padding=True, truncation=True, max_length=512)
+    translated = model.generate(**inputs)
+    result = tokenizer.decode(translated[0], skip_special_tokens=True)
+    return result
 
-# Thiết lập giao diện Gradio
-interface = gr.Interface(
-    fn=translate_zh_to_vi,
+demo = gr.Interface(
+    fn=translate,
     inputs=gr.Textbox(
-        lines=5, 
-        label="Nhập văn bản tiếng Trung (Chinese)", 
-        placeholder="Ví dụ: 你好，欢迎来到越南！"
+        lines=5,
+        placeholder="Nhập văn bản tiếng Trung...",
+        label="📝 Văn bản tiếng Trung (中文)"
     ),
     outputs=gr.Textbox(
-        lines=5, 
-        label="Bản dịch tiếng Việt (Vietnamese)"
+        lines=5,
+        label="🇻🇳 Bản dịch tiếng Việt"
     ),
-    title="Trình dịch Trung - Việt (Helsinki-NLP)",
-    description="Ứng dụng dịch thuật sử dụng mô hình mã nguồn mở `Helsinki-NLP/opus-mt-zh-vi`.",
-    theme="default",
+    title="🈶 Dịch Trung → Việt",
+    description="Sử dụng model Helsinki-NLP/opus-mt-zh-vi để dịch văn bản từ tiếng Trung sang tiếng Việt.",
+    examples=[
+        ["你好，世界！"],
+        ["今天天气很好。"],
+        ["我想学习越南语。"],
+    ],
+    flagging_mode="never"
 )
 
-# Khởi chạy ứng dụng
-if __name__ == "__main__":
-    interface.launch()
+demo.launch()