“Update”

config/config.py

@@ -4,11 +4,11 @@ from config.prompts import prompts  # Import prompts from prompts.py
 from config.models import models
 
 # Retrieve the Hugging Face token
-#pi_token = os.getenv("HF_CTB_TOKEN")
+api_token = os.getenv("HF_CTB_TOKEN")
 
 # Debugging: Check if the Hugging Face token is available
 if not api_token:
-    print("ERROR: Hugging Face token (HF_CTB_TOKEN) is missing. Please set it as an environment variable or in Colab secrets.")
+    print("ERROR1: Hugging Face token (HF_CTB_TOKEN) is missing. Please set it as an environment variable or in Colab secrets.")
 else:
     print("Hugging Face token loaded successfully.")
 

src/img_gen.py

@@ -20,7 +20,7 @@ def generate_image(prompt_alias, team, model_alias, custom_prompt, height=360, w
     api_token = os.getenv("HF_CTB_TOKEN")
 
     if not api_token:
-        return None, "ERROR: Hugging Face token (HF_CTB_TOKEN) is missing. Please set it as an environment variable."
+        return None, "ERROR2: Hugging Face token (HF_CTB_TOKEN) is missing. Please set it as an environment variable."
 
     # Find the selected prompt and model
     try: