Spaces:
Running
Running
Commit
·
5c991e7
1
Parent(s):
7740657
Default lang to Portuguese
Browse files
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=
|
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(
|
|
|
|
|
|
|
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}')
|