|
'''Test EasyNMT opus-mt |
|
|
|
''' |
|
|
|
import streamlit as st |
|
from easynmt import EasyNMT |
|
model = EasyNMT('opus-mt') |
|
|
|
|
|
|
|
|
|
|
|
|
|
st.set_page_config( |
|
page_title='EasyNMT Testing', |
|
page_icon='๐', |
|
layout='wide', |
|
) |
|
|
|
|
|
with st.container(): |
|
st.markdown('## ๐ Machine Translation') |
|
st.write('This is a testing of EasyNMT and opus-mt model with Streamlit.') |
|
|
|
lang_list = model.get_languages() |
|
|
|
b_size = st.slider('Translation quality (Beam size)', 1, 10, 5) |
|
|
|
text = '' |
|
submit = '' |
|
target_langs = '' |
|
|
|
with st.form(key='nmt'): |
|
text = st.text_area( |
|
label='Enter text', |
|
|
|
help='Auto detect input language 170+.') |
|
target_langs = st.multiselect( |
|
'Translate to (can select more than one language)', |
|
lang_list, |
|
['en', 'vi']) |
|
submit = st.form_submit_button(label='Translate') |
|
|
|
if submit and text: |
|
detected_lang = model.language_detection(text) |
|
|
|
st.write('Dectected input language: ' + detected_lang) |
|
|
|
if not target_langs: |
|
st.error('Please choose at least 1 target language.') |
|
st.stop() |
|
|
|
for lang in target_langs: |
|
try: |
|
res = model.translate(text, target_lang=lang, beam_size=b_size) |
|
if res: |
|
st.success(lang + ' => ' + res) |
|
|
|
except Exception as e: |
|
st.write(e) |
|
|
