app.py

import streamlit as st
from transformers import pipeline, AutoTokenizer

tokenizer = AutoTokenizer.from_pretrained("Helsinki-NLP/opus-mt-vi-en")

translator = pipeline("translation_vi_to_en", model="Helsinki-NLP/opus-mt-vi-en")

st.title("Demo translation with pretrained model")

input_text = st.text_area("Nhập văn bản tiếng Việt:", "")

if input_text:
    encoded_text = tokenizer(input_text, return_tensors="pt", padding=True, truncation=True)

    translated_text = translator(input_text)

    st.write("Translation Vietnamese to English:")
    st.write(translated_text[0]["translation_text"])