linoyts HF Staff commited on
Commit
c4ad0a8
·
verified ·
1 Parent(s): 631c9cd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -270,7 +270,7 @@ def infer(input_image, prompt, illumination_dropdown, direction_dropdown, seed=4
270
  prompt = ILLUMINATION_OPTIONS[illumination_dropdown]
271
 
272
  #If the prompt matches the illumination options, prefix that
273
- if prompt == ILLUMINATION_OPTIONS[illumination_dropdown]:
274
  prompt_prefix = f", with {illumination_dropdown}"
275
  #If the prompt was changed, the prefix is empty as the user prompt is predominant
276
  else:
 
270
  prompt = ILLUMINATION_OPTIONS[illumination_dropdown]
271
 
272
  #If the prompt matches the illumination options, prefix that
273
+ if illumination_dropdown != "custom" and prompt == ILLUMINATION_OPTIONS[illumination_dropdown]:
274
  prompt_prefix = f", with {illumination_dropdown}"
275
  #If the prompt was changed, the prefix is empty as the user prompt is predominant
276
  else: