Update app.py
Browse files
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 =
|
|
|
|
|
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)
|