danilommarano commited on
Commit
7740657
·
1 Parent(s): 1bda44e

Add title and introduction text

Browse files
Files changed (1) hide show
  1. app.py +9 -4
app.py CHANGED
@@ -1,10 +1,15 @@
1
  import streamlit as st
2
  from transformers import MBartForConditionalGeneration, MBart50TokenizerFast
3
 
4
- model = MBartForConditionalGeneration.from_pretrained("facebook/mbart-large-50-one-to-many-mmt")
5
- tokenizer = MBart50TokenizerFast.from_pretrained("facebook/mbart-large-50-one-to-many-mmt", src_lang="en_XX")
6
-
7
- import streamlit as st
 
 
 
 
 
8
 
9
  language = st.selectbox(
10
  "Lingua de saída: ",
 
1
  import streamlit as st
2
  from transformers import MBartForConditionalGeneration, MBart50TokenizerFast
3
 
4
+ model_name = "facebook/mbart-large-50-one-to-many-mmt"
5
+ model = MBartForConditionalGeneration.from_pretrained(model_name)
6
+ tokenizer = MBart50TokenizerFast.from_pretrained(model_name, src_lang="en_XX")
7
+
8
+ st.title("Tradutor de Inglês")
9
+ st.write(
10
+ f"Faça a tradução do seu texto em inglês para "
11
+ f"{len(tokenizer.lang_code_to_id.keys())}!"
12
+ )
13
 
14
  language = st.selectbox(
15
  "Lingua de saída: ",