Create app.py

app.py

@@ -0,0 +1,29 @@
+import streamlit as st
+from transformers import pipeline
+
+# Tải pipeline với mô hình sửa lỗi chính tả của bạn
+@st.cache_resource
+def load_corrector():
+    return pipeline("text2text-generation", model="hHoai/model_vietnamcorrection")  # Thay "username/checkpoint-190" bằng đường dẫn đến mô hình của bạn
+
+corrector = load_corrector()
+MAX_LENGTH = 512
+
+# Tiêu đề ứng dụng
+st.title("Demo Sửa Lỗi Chính Tả")
+st.write("Nhập vào văn bản tiếng Việt có lỗi, và công cụ sẽ đưa ra phiên bản đã được chỉnh sửa.")
+
+# Nhập liệu từ người dùng
+input_text = st.text_area("Nhập văn bản:", placeholder="Ví dụ: Kính chúc Bác, Mr.Trường va em Dật và các đồng bào yên hảo.", height=150)
+
+if st.button("Sửa lỗi"):
+    if input_text.strip():
+        # Sửa lỗi chính tả
+        predictions = corrector([input_text], max_length=MAX_LENGTH)
+        corrected_text = predictions[0]["generated_text"]
+        
+        # Hiển thị kết quả
+        st.subheader("Văn bản đã sửa:")
+        st.write(corrected_text)
+    else:
+        st.warning("Vui lòng nhập văn bản để sửa lỗi!")