Spaces:
Runtime error
Runtime error
Mel Nguyen (she/her)
commited on
Commit
•
f546c1a
1
Parent(s):
fd135e5
Add initial files
Browse files- app.py +32 -0
- requirements.txt +3 -0
app.py
ADDED
@@ -0,0 +1,32 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import transformers
|
2 |
+
import streamlit as st
|
3 |
+
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
|
4 |
+
|
5 |
+
|
6 |
+
st.title("English-Vietnamese Text Translator")
|
7 |
+
st.write("A simple interface to translate from English to Vietnamese, and vice versa.")
|
8 |
+
|
9 |
+
model = AutoModelForSeq2SeqLM.from_pretrained("facebook/nllb-200-distilled-600M")
|
10 |
+
tokenizer = AutoTokenizer.from_pretrained("facebook/nllb-200-distilled-600M")
|
11 |
+
|
12 |
+
default_value = "UN Chief says there is no military solution in Syria"
|
13 |
+
sent = st.text_area("Input Your Text Here", default_value, height = 275)
|
14 |
+
src_lang_selection = st.radio(
|
15 |
+
"Select Your Source Language:",
|
16 |
+
('English', 'Vietnamese'))
|
17 |
+
|
18 |
+
if src_lang_selection == "English":
|
19 |
+
src_lang = "eng_Latn"
|
20 |
+
tgt_lang = "vie_Latn"
|
21 |
+
else:
|
22 |
+
src_lang = "vie_Latn"
|
23 |
+
tgt_lang = "eng_Latn"
|
24 |
+
|
25 |
+
if st.button("Run"):
|
26 |
+
with st.spinner("Working Hard..."):
|
27 |
+
translator = pipeline('translation', model=model, tokenizer=tokenizer, src_lang=src_lang, tgt_lang=tgt_lang)
|
28 |
+
trans_text = translator(sent)
|
29 |
+
st.write(trans_text)
|
30 |
+
st.success("Done!")
|
31 |
+
|
32 |
+
st.write("For feedback/requests, write to mel.nguyen273@gmail.com.")
|
requirements.txt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
transformers
|
2 |
+
streamlit
|
3 |
+
torch
|