ajuste

app.py

@@ -22,11 +22,11 @@ def carregar_modelo_e_tokenizador_mbart(modelo):
 
 # TODO:batch?
 def traduzir_en_pt(text):
-    inputs = tokenizer(text, return_tensors='pt')
+    inputs = tokenizador(text, return_tensors='pt')
     input_ids = inputs.input_ids
     attention_mask = inputs.attention_mask
-    output = model.generate(input_ids, attention_mask=attention_mask, forced_bos_token_id=tokenizer.lang_code_to_id['pt_XX'])
-    return tokenizer.decode(output[0], skip_special_tokens=True)
+    output = modelo.generate(input_ids, attention_mask=attention_mask, forced_bos_token_id=tokenizador.lang_code_to_id['pt_XX'])
+    return tokenizador.decode(output[0], skip_special_tokens=True)
 
 ## streamlit ##
 def carregar_dataset():
@@ -77,9 +77,9 @@ dataset = carregar_dataset()
 if dataset is not None:
     mostrar_dataset()
     if st.button("Carregar modelo"):
-        model, tokenizer = carregar_modelo()
+        modelo, tokenizador = carregar_modelo()
 
 
-if st.button("Traduzir dataset") and model is not None:
+if st.button("Traduzir dataset") and modelo is not None:
     traduzir_dataset(dataset)
     resultado()