Update app.py

app.py

@@ -1,14 +1,22 @@
-from transformers import pipeline
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 import gradio as gr
 
-translator = pipeline(
-    "translation_ru_to_vi",
-    model="Helsinki-NLP/opus-mt-ru-vi"
-)
+model_name = "Helsinki-NLP/opus-mt-ru-vi"
+
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
 
 def translate(text):
-    result = translator(text)
-    return result[0]["translation_text"]
+    inputs = tokenizer(text, return_tensors="pt")
+
+    outputs = model.generate(**inputs)
+
+    translated = tokenizer.decode(
+        outputs[0],
+        skip_special_tokens=True
+    )
+
+    return translated
 
 iface = gr.Interface(
     fn=translate,
@@ -16,9 +24,8 @@ iface = gr.Interface(
         lines=2,
         placeholder="Nhập tiếng Nga..."
     ),
-    outputs=gr.Textbox(),
-    title="🇷🇺 Nga → Việt",
-    description="Mini translator for Notion"
+    outputs="text",
+    title="🇷🇺 Nga → Việt"
 )
 
 iface.launch(server_name="0.0.0.0")