Update app.py

app.py

@@ -7,6 +7,10 @@ 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
+# 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"
@@ -21,11 +25,16 @@ def translation(source, target, text):
     #start_time = time.time()
     source = flores_codes[source]
     target = flores_codes[target]
-
-    translator = pipeline('translation', model=model, tokenizer=tokenizer, 
-
-    src_lang=source, tgt_lang=target)
-    output = translator(text, max_length=400)
+    if target == flores_codes["Asturianu"]:
+        texto_castellano = source
+        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)
+        output = translator_ast(text, max_length=400)
+    else
+        translator = pipeline('translation', model=model, tokenizer=tokenizer, src_lang=source, tgt_lang=target)
+        output = translator(text, max_length=400)
 
     #end_time = time.time()