Update app.py

app.py

@@ -6,9 +6,9 @@ import gradio as gr
 from datetime import datetime
 
 # Debugging: Check if the Hugging Face token is available
-api_token = os.getenv("HF_TOKEN")
+api_token = os.getenv("HF_CTB_TOKEN")
 if not api_token:
-    print("ERROR: Hugging Face token (HF_TOKEN) is missing. Please set it as an environment variable.")
+    print("ERROR: Hugging Face token (HF_CTB_TOKEN) is missing. Please set it as an environment variable.")
 else:
     print("Hugging Face token loaded successfully.")
 
@@ -67,7 +67,7 @@ prompts = [
 def generate_image(prompt_alias, team, model_alias, height, width, num_inference_steps, guidance_scale, seed):
     # Debugging: Check if the token is available
     if not api_token:
-        return None, "ERROR: Hugging Face token (HF_TOKEN) is missing. Please set it as an environment variable."
+        return None, "ERROR: Hugging Face token (HF_CTB_TOKEN) is missing. Please set it as an environment variable."
 
     # Find the selected prompt and model
     prompt = next(p for p in prompts if p["alias"] == prompt_alias)["text"]