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

Default lang to Portuguese

Browse files
Files changed (1) hide show
  1. app.py +6 -3
app.py CHANGED
@@ -8,13 +8,13 @@ tokenizer = MBart50TokenizerFast.from_pretrained(model_name, src_lang="en_XX")
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: ",
16
  tokenizer.lang_code_to_id.keys(),
17
- index=0,
18
  placeholder="Selecione uma linguagem...",
19
  )
20
 
@@ -27,6 +27,9 @@ generated_tokens = model.generate(
27
  forced_bos_token_id=tokenizer.lang_code_to_id[language]
28
  )
29
 
30
- translation = tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)
 
 
 
31
 
32
  st.write(f'Resultado: {translation}')
 
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())} linguagens!"
12
  )
13
 
14
  language = st.selectbox(
15
  "Lingua de saída: ",
16
  tokenizer.lang_code_to_id.keys(),
17
+ index=40,
18
  placeholder="Selecione uma linguagem...",
19
  )
20
 
 
27
  forced_bos_token_id=tokenizer.lang_code_to_id[language]
28
  )
29
 
30
+ translation = tokenizer.batch_decode(
31
+ generated_tokens,
32
+ skip_special_tokens=True
33
+ )
34
 
35
  st.write(f'Resultado: {translation}')