Spaces:
Sleeping
Sleeping
Andre
commited on
Commit
·
1700c38
1
Parent(s):
6cb74a2
“Update”
Browse files- src/img_gen.py +0 -6
src/img_gen.py
CHANGED
@@ -16,12 +16,6 @@ def generate(prompt_alias, team, model_alias, custom_prompt, height=360, width=6
|
|
16 |
|
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:
|
27 |
prompt = next(p for p in prompts if p["alias"] == prompt_alias)["text"]
|
|
|
16 |
|
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 |
# Find the selected prompt and model
|
20 |
try:
|
21 |
prompt = next(p for p in prompts if p["alias"] == prompt_alias)["text"]
|