Update app.py

app.py

@@ -1,22 +1,33 @@
 import gradio as gr
 from transformers import pipeline
 
-# Cargar el modelo de traducción
-translator = pipeline("translation_es_to_ast", model="Helsinki-NLP/opus-mt-es-ast")
+# Inicializar el modelo de traducción
+translator = pipeline("translation", model="projecte-aina/aina-translator-es-ast")
 
-# Función para traducir texto
-def translate(text):
-    translation = translator(text, max_length=400)
-    return translation[0]['translation_text']
+def translate_text(text):
+    """
+    Traduce texto del español al asturiano
+    """
+    if not text:
+        return ""
+    
+    result = translator(text, max_length=400)
+    return result[0]['translation_text']
 
-# Crear la interfaz de Gradio
-iface = gr.Interface(
-    fn=translate,
-    inputs=gr.Textbox(lines=2, placeholder="Introduce el texto en español aquí..."),
-    outputs="text",
-    title="Traductor Español - Asturiano",
-    description="Traduce texto de español a asturiano utilizando el modelo Helsinki-NLP/opus-mt-es-ast."
+# Crear la interfaz Gradio
+demo = gr.Interface(
+    fn=translate_text,
+    inputs=gr.Textbox(label="Texto en español", placeholder="Escribe aquí el texto a traducir..."),
+    outputs=gr.Textbox(label="Traducción al asturiano"),
+    title="Traductor Español-Asturiano",
+    description="Traductor basado en el modelo AINA para traducir del español al asturiano.",
+    examples=[
+        ["Hola, ¿cómo estás?"],
+        ["Me gusta mucho Asturias y su cultura"],
+        ["El cielo está muy azul hoy"]
+    ]
 )
 
 # Lanzar la aplicación
-iface.launch()
+if __name__ == "__main__":
+    demo.launch()