Mel Nguyen (she/her) commited on
Commit
f546c1a
1 Parent(s): fd135e5

Add initial files

Browse files
Files changed (2) hide show
  1. app.py +32 -0
  2. 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