Update app.py

app.py

@@ -7,7 +7,7 @@ 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 fine-tuned que traduce del español
+# Pero como traduce mal al asturiano, añadimos este otro fine-tuned
 # Es del proyecto AINA: https://huggingface.co/projecte-aina
 model_ast = "projecte-aina/aina-translator-es-ast"
 
@@ -24,16 +24,12 @@ def translation(source, target, text):
     #start_time = time.time()
     source = flores_codes[source]
     target = flores_codes[target]
-    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, src_lang=flores_codes["Castellano"], tgt_lang=flores_codes["Asturianu"])
-        output = translator_ast(text, max_length=400)
+    if source == flores_codes["Asturianu"] or target == flores_codes["Asturianu"]:
+        translator = pipeline('translation', model=model_ast, tokenizer=tokenizer, src_lang=source, tgt_lang=target)
     else:
         translator = pipeline('translation', model=model, tokenizer=tokenizer, src_lang=source, tgt_lang=target)
-        output = translator(text, max_length=400)
+ 
+    output = translator(text, max_length=400)
 
     #end_time = time.time()