Update app.py

app.py

@@ -7,10 +7,9 @@ from transformers import AutoTokenizer, pipeline
 # Modelo de Meta capaz de traducir a más de 200 idiomas
 model = 'facebook/nllb-200-distilled-600M'
 tokenizer = model
-# Pero como traduce mal al asturiano, añadimos este otro que traduce del español
+# Pero como traduce mal al asturiano, añadimos este otro fine-tuned que traduce del español
 # Es del proyecto AINA: https://huggingface.co/projecte-aina
 model_ast = "projecte-aina/aina-translator-es-ast"
-tokenizer_ast = model_ast
 
 flores_codes = {}
 flores_codes["Asturianu"] = "ast_Latn"
@@ -30,7 +29,7 @@ def translation(source, target, text):
         if source != flores_codes["Castellano"]:
             translator = pipeline('translation', model=model, tokenizer=tokenizer, src_lang=source, tgt_lang=flores_codes["Castellano"])
             texto_castellano = translator(text, max_length=400)
-        translator_ast = pipeline('translation', model=model_ast, tokenizer=tokenizer_ast)
+        translator_ast = pipeline('translation', model=model_ast, tokenizer=tokenizer)
         output = translator_ast(text, max_length=400)
     else:
         translator = pipeline('translation', model=model, tokenizer=tokenizer, src_lang=source, tgt_lang=target)
@@ -73,5 +72,6 @@ if __name__ == '__main__':
                  outputs,
                  title=title,
                  description=description,
+                 submit_btn="Traducir"
                  ).launch()