Update app.py

app.py

@@ -1,23 +1,26 @@
 from transformers import pipeline
 import gradio as gr
+import os  # Importa la librería 'os' para acceder a las variables de entorno
 
-# Define el modelo de traducción inglés-español
+# Modelo de traducción inglés-español (no requiere token para la mayoría de los casos)
 modelo_traduccion = pipeline("translation", model="Helsinki-NLP/opus-mt-en-es")
 
-# Define el modelo de clasificación de idiomas (cambiado a joosbu/roberta-small-langdet)
-modelo_idioma = pipeline("text-classification", model="joosbu/roberta-small-langdet")
-
 def detectar_idioma(texto):
     """
-    Detecta el idioma del texto utilizando el modelo roberta-small-langdet.
+    Detecta el idioma del texto utilizando el modelo xlm-roberta-large, leyendo el token del entorno.
 
     Args:
         texto (str): El texto del cual se quiere detectar el idioma.
 
     Returns:
-        str: El código del idioma detectado (ej: 'en', 'es').
+        str: El código del idioma detectado (ej: 'en', 'es') o None si hay un error.
     """
+    api_token = os.environ.get("HF_TOKEN")  # Lee el valor de la variable de entorno HF_TOKEN
     try:
+        kwargs = {}
+        if api_token:
+            kwargs['token'] = api_token
+        modelo_idioma = pipeline("text-classification", model="papluca/xlm-roberta-large-language-detection", **kwargs)
         resultado = modelo_idioma(texto, top_k=1)[0]
         return resultado['label']
     except Exception as e:
@@ -26,7 +29,7 @@ def detectar_idioma(texto):
 
 def traducir_texto(texto_en):
     """
-    Traduce el texto de entrada del inglés al español, previa verificación del idioma.
+    Traduce el texto de entrada del inglés al español, leyendo el token del entorno.
 
     Args:
         texto_en (str): El texto en inglés que se va a traducir.
@@ -62,7 +65,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 (roberta-small-langdet).
+    Define la interfaz gráfica de la aplicación de traducción de inglés a español con lectura del token desde el entorno.
     """
     gr.Markdown("# Traductor Inglés a Español")
     with gr.Row():