Language option select box

app.py

@@ -4,13 +4,22 @@ from transformers import MBartForConditionalGeneration, MBart50TokenizerFast
 model = MBartForConditionalGeneration.from_pretrained("facebook/mbart-large-50-one-to-many-mmt")
 tokenizer = MBart50TokenizerFast.from_pretrained("facebook/mbart-large-50-one-to-many-mmt", src_lang="en_XX")
 
-input_text = st.text_area("")
+import streamlit as st
+
+option = st.selectbox(
+   "Lingua de saída: ",
+   (tokenizer.lang_code_to_id.keys()),
+   index=None,
+   placeholder="Selecione uma linguagem...",
+)
+
+input_text = st.text_area("Texto de entrada:")
 
 model_inputs = tokenizer(input_text, return_tensors="pt")
 
 generated_tokens = model.generate(
     **model_inputs,
-    forced_bos_token_id=tokenizer.lang_code_to_id["pt_BR"]
+    forced_bos_token_id=tokenizer.lang_code_to_id[option]
 )
 
 translation = tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)