Update app.py

app.py

@@ -1,6 +1,16 @@
 import streamlit as st
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
+import os
+from huggingface_hub import login
+
+# Iniciar sesión en Hugging Face utilizando un token guardado en una variable de entorno
+def login_to_huggingface():
+    read_key = os.getenv("Read_Token")  # Asegúrate de tener la variable de entorno configurada
+    if read_key:
+        login(token=read_key)
+    else:
+        st.error("No se encontró el token de Hugging Face. Asegúrate de que la variable de entorno 'HUGGINGFACE_TOKEN' esté configurada.")
 
 # Inicializar el modelo generador de texto
 @st.cache_resource  # Cachear el modelo para no recargarlo repetidamente
@@ -14,6 +24,9 @@ def load_model():
     )
     return tokenizer, model
 
+# Iniciar sesión
+login_to_huggingface()
+
 # Cargar el modelo y el tokenizador
 tokenizer, model = load_model()
 
@@ -45,4 +58,4 @@ if st.button("Generar texto"):
         st.subheader("Texto generado:")
         st.write(generated_text)
     else:
-        st.error("Por favor, introduce un prompt antes de generar texto.")
+        st.error("Por favor, introduce un prompt antes de generar texto.")