Update app.py

app.py

@@ -7,16 +7,16 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 model_id = "somosnlp-hackathon-2025/leia_preference_model_social_norms"
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
-# Obtener el token secreto del entorno
+# Obtener el token desde variable de entorno
 hf_token = os.environ.get("SSNZHY_TOKEN")
+if not hf_token:
+    raise ValueError("El token de Hugging Face no está definido. Configura la variable de entorno 'SSNZHY_TOKEN'.")
 
 # Cargar modelo y tokenizer con autenticación
 tokenizer = AutoTokenizer.from_pretrained(model_id, token=hf_token)
 model = AutoModelForCausalLM.from_pretrained(model_id, token=hf_token).to(device)
 
-
-
-# Función para responder
+# Función de respuesta del chatbot
 def respond(message, history, system_message, max_tokens, temperature, top_p):
     prompt = system_message + "\n"
     for user, assistant in history:
@@ -39,6 +39,7 @@ def respond(message, history, system_message, max_tokens, temperature, top_p):
 
 
 
+
 # Descripción del proyecto
 descripcion = """
 
@@ -100,20 +101,32 @@ En **LeIA GO**, somos conscientes de que los modelos de lenguaje pueden reflejar
 - Dataset: [https://huggingface.co/datasets/somosnlp-hackathon-2025/dataset-preferencias-v0] 
 """
 
-# Interfaz Gradio
+# Interfaz con Gradio
 with gr.Blocks() as demo:
     with gr.Tab("Descripción del Proyecto"):
         gr.Markdown(descripcion)
+
     with gr.Tab("Chatbot LeIA GO"):
+        gr.Markdown("### 👋 Bienvenida a LeIA GO\nChatea con una IA especializada en normas sociales y variantes del español.")
         gr.ChatInterface(
             respond,
             additional_inputs=[
-                gr.Textbox(value="Eres una asistente lingüística especializada en español regional.", label="System message"),
+                gr.Textbox(
+                    value="Eres una asistente lingüística especializada en español regional.",
+                    label="Mensaje del sistema"
+                ),
                 gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
                 gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
                 gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p (nucleus sampling)"),
             ],
         )
 
+# Lanzamiento
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(
+        show_api=False,
+        share=False,
+        title="LeIA GO - Chat de Normas Sociales",
+        theme=gr.themes.Base()
+    )
+