Update app.py

app.py

@@ -8,6 +8,13 @@ from huggingface_hub import InferenceClient
 
 st.set_page_config(layout="wide")
 
+# Configuración para obtener el token de Hugging Face desde las variables de entorno
+HF_TOKEN = os.getenv("HF_TOKEN")
+
+if not HF_TOKEN:
+    st.error("El token de Hugging Face no está configurado. Asegúrate de definir la variable de entorno 'HF_TOKEN'.")
+    st.stop()  # Detener ejecución si no hay token
+
 try:
     with open("config.yaml", "r") as file: 
         credentials = yaml.safe_load(file)
@@ -21,7 +28,6 @@ class AppConfig:
     CLEANUP_DAYS: int = 7    
 
 MAX_SEED = AppConfig.MAX_SEED
-HF_TOKEN = os.environ.get("HF_TOKEN")
 DATA_PATH = Path("./data")
 DATA_PATH.mkdir(exist_ok=True)
 
@@ -223,4 +229,4 @@ def main():
     display_gallery()
 
 if __name__ == "__main__": 
-    main()
+    main()