Update app.py

Cambio de modelo de clasificacion

app.py

@@ -4,12 +4,12 @@ import gradio as gr
 # Define el modelo de traducción inglés-español
 modelo_traduccion = pipeline("translation", model="Helsinki-NLP/opus-mt-en-es")
 
-# Define el modelo de clasificación de idiomas
-modelo_idioma = pipeline("text-classification", model="papluca/xlm-roberta-large-language-detection")
+# Define el modelo de clasificación de idiomas (cambiado a fasttext)
+modelo_idioma = pipeline("text-classification", model="facebook/fasttext-language-identification")
 
 def detectar_idioma(texto):
     """
-    Detecta el idioma del texto utilizando un modelo de clasificación.
+    Detecta el idioma del texto utilizando el modelo fasttext.
 
     Args:
         texto (str): El texto del cual se quiere detectar el idioma.
@@ -19,7 +19,8 @@ def detectar_idioma(texto):
     """
     try:
         resultado = modelo_idioma(texto, top_k=1)[0]
-        return resultado['label']
+        # El modelo fasttext devuelve códigos de idioma precedidos por '__label__'
+        return resultado['label'].replace('__label__', '')
     except Exception as e:
         print(f"Error en la detección de idioma: {e}")
         return None
@@ -62,7 +63,7 @@ def borrar_entrada(texto_entrada):
 
 with gr.Blocks() as interfaz_traductor:
     """
-    Define la interfaz gráfica de la aplicación de traducción de inglés a español con detección de idioma y título estándar.
+    Define la interfaz gráfica de la aplicación de traducción de inglés a español con detección de idioma (fasttext).
     """
     gr.Markdown("# Traductor Inglés a Español")
     with gr.Row():