Update app.py

app.py

@@ -1,15 +1,19 @@
 import gradio as gr
 import torch
+import os
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
 # Configuración del modelo
 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
+hf_token = os.environ.get("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)
 
-# Cargar modelo y tokenizer
-tokenizer = AutoTokenizer.from_pretrained(model_id)
-model = AutoModelForCausalLM.from_pretrained(model_id).to(device)
 
 
 # Función para responder