Andre commited on
Commit
e22b115
·
1 Parent(s): 57af060

“Update”

Browse files
Files changed (1) hide show
  1. src/img_gen.py +3 -3
src/img_gen.py CHANGED
@@ -17,10 +17,10 @@ def generate(prompt_alias, team, model_alias, custom_prompt, height=360, width=6
17
 
18
  def generate_image(prompt_alias, team, model_alias, custom_prompt, height=360, width=640, num_inference_steps=20, guidance_scale=2.0, seed=-1):
19
  # Debugging: Check if the token is available
20
- api_token = os.getenv("HF_CTB_TOKEN")
21
 
22
- if not api_token:
23
- return None, "ERROR2: Hugging Face token (HF_CTB_TOKEN) is missing. Please set it as an environment variable."
24
 
25
  # Find the selected prompt and model
26
  try:
 
17
 
18
  def generate_image(prompt_alias, team, model_alias, custom_prompt, height=360, width=640, num_inference_steps=20, guidance_scale=2.0, seed=-1):
19
  # Debugging: Check if the token is available
20
+ #api_token = os.getenv("HF_CTB_TOKEN")
21
 
22
+ #if not api_token:
23
+ #return None, "ERROR2: Hugging Face token (HF_CTB_TOKEN) is missing. Please set it as an environment variable."
24
 
25
  # Find the selected prompt and model
26
  try: