“Update”

app.py

@@ -2,6 +2,7 @@
 import sys
 import os
 
+
 # Add the src folder to the Python path
 # Solves all problems w subfolders - option2
 src_path = os.path.abspath(os.path.join(os.path.dirname(__file__), "src"))
@@ -10,12 +11,6 @@ if src_path not in sys.path:
 
 # Import gradio_interface
 from gradio_interface import demo
-from config.config import api_token
-from huggingface_hub import InferenceClient
-from config.models import models
-
-# Initialize the InferenceClient with the default model
-client = InferenceClient(models[0]["name"], token=api_token)
 
 
 if __name__ == "__main__":

config/config.py

@@ -4,7 +4,7 @@ from config.prompts import prompts  # Import prompts from prompts.py
 from config.models import models
 
 # Retrieve the Hugging Face token
-api_token = os.getenv("HF_CTB_TOKEN")
+#pi_token = os.getenv("HF_CTB_TOKEN")
 
 # Debugging: Check if the Hugging Face token is available
 if not api_token:

src/img_gen.py

@@ -4,8 +4,7 @@ import os
 import random
 from huggingface_hub import InferenceClient
 from datetime import datetime
-from config.config import api_token, models, prompts  # Direct import
-
+from config.config import models, prompts  # Direct import
 
 def generate(prompt_alias, team, model_alias, custom_prompt, height=360, width=640, num_inference_steps=20, guidance_scale=2.0, seed=-1):
     try:
@@ -18,6 +17,8 @@ def generate(prompt_alias, team, model_alias, custom_prompt, height=360, width=6
 
 def generate_image(prompt_alias, team, model_alias, custom_prompt, height=360, width=640, num_inference_steps=20, guidance_scale=2.0, seed=-1):
     # Debugging: Check if the token is available
+    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."