Update app.py

app.py

@@ -1,5 +1,17 @@
+import os
+from dotenv import load_dotenv
 import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
+from huggingface_hub import login
+
+# Cargar las variables de entorno desde el archivo .env
+load_dotenv()
+
+# Iniciar sesión en Hugging Face usando la variable de entorno
+huggingface_token = os.getenv('HUGGINGFACE_TOKEN')
+if huggingface_token is None:
+    raise ValueError("El token de Hugging Face no está configurado en las variables de entorno.")
+login(huggingface_token)
 
 # Cargar el modelo y el tokenizador
 model_name = "meta-llama/Llama-3.2-1B-Instruct"  # Asegúrate de que este sea el nombre correcto