LeeveWasTaken commited on
Commit
3bec76d
·
verified ·
1 Parent(s): 4024048

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -29,7 +29,9 @@ def update_imgbox(choices):
29
  def gen_fn(model_str, prompt, negative_prompt=None, image_style="Default"):
30
  if model_str == 'NA':
31
  return None
32
- modified_prompt = f"{prompt}, {image_style}"
 
 
33
  if negative_prompt:
34
  modified_prompt += f", not {negative_prompt}"
35
  return models_load[model_str](modified_prompt)
@@ -81,4 +83,4 @@ with gr.Blocks() as demo:
81
  """
82
  )
83
 
84
- demo.launch(max_threads=200)
 
29
  def gen_fn(model_str, prompt, negative_prompt=None, image_style="Default"):
30
  if model_str == 'NA':
31
  return None
32
+ modified_prompt = prompt
33
+ if image_style != "Default":
34
+ modified_prompt += f", {image_style}"
35
  if negative_prompt:
36
  modified_prompt += f", not {negative_prompt}"
37
  return models_load[model_str](modified_prompt)
 
83
  """
84
  )
85
 
86
+ demo.launch(max_threads=200)