Spaces:
Sleeping
Sleeping
Andre
commited on
Commit
·
e22b115
1
Parent(s):
57af060
“Update”
Browse files- 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:
|