Update app.py

app.py

@@ -2,11 +2,12 @@ import streamlit as st
 from transformers import pipeline
 
 pipe = pipeline(model="thainq107/en-vi-mbart50")
-st.title("Dự đoán với mô hình HuggingFace")
+st.title("Dịch với mô hình HuggingFace")
 user_input = st.text_area("Nhập đoạn văn tiếng Anh:", "")
-if st.button("Dự đoán"):
+if st.button("Dịch"):
     if user_input:
-        result = pipe(user_input)
-        st.write("Kết quả dự đoán:", result)
+        translated_text = pipe("I go to school", num_beams=1, do_sample=False)
+        result = translated_text.generated_text
+        st.write("Kết quả dịch:", result)
     else:
-        st.write("Vui lòng nhập đoạn văn để dự đoán.")
+        st.write("Vui lòng nhập đoạn văn để dịch.")