Update app.py
Browse filesdropdown prompt
app.py
CHANGED
|
@@ -83,6 +83,8 @@ pipe.to(device)
|
|
| 83 |
|
| 84 |
@spaces.GPU()
|
| 85 |
def generate_image(
|
|
|
|
|
|
|
| 86 |
upload_images,
|
| 87 |
prompt,
|
| 88 |
negative_prompt,
|
|
@@ -112,6 +114,15 @@ def generate_image(
|
|
| 112 |
adapter_conditioning_factor = 0.
|
| 113 |
sketch_image = None
|
| 114 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 115 |
# check the trigger word
|
| 116 |
image_token_id = pipe.tokenizer.convert_tokens_to_ids(pipe.trigger_word)
|
| 117 |
input_ids = pipe.tokenizer.encode(prompt)
|
|
@@ -502,8 +513,6 @@ with gr.Blocks(css=css) as demo:
|
|
| 502 |
value=default_prompt_keys[0],
|
| 503 |
interactive=True
|
| 504 |
)
|
| 505 |
-
|
| 506 |
-
|
| 507 |
|
| 508 |
|
| 509 |
gender_selector = gr.Radio(
|
|
|
|
| 83 |
|
| 84 |
@spaces.GPU()
|
| 85 |
def generate_image(
|
| 86 |
+
prompt_name,
|
| 87 |
+
gender,
|
| 88 |
upload_images,
|
| 89 |
prompt,
|
| 90 |
negative_prompt,
|
|
|
|
| 114 |
adapter_conditioning_factor = 0.
|
| 115 |
sketch_image = None
|
| 116 |
|
| 117 |
+
|
| 118 |
+
#added
|
| 119 |
+
prompt_dict = get_prompts_by_gender(gender)
|
| 120 |
+
|
| 121 |
+
# added
|
| 122 |
+
actual_prompt = prompt_dict.get(prompt_name, "")
|
| 123 |
+
prompt = actual_prompt
|
| 124 |
+
|
| 125 |
+
|
| 126 |
# check the trigger word
|
| 127 |
image_token_id = pipe.tokenizer.convert_tokens_to_ids(pipe.trigger_word)
|
| 128 |
input_ids = pipe.tokenizer.encode(prompt)
|
|
|
|
| 513 |
value=default_prompt_keys[0],
|
| 514 |
interactive=True
|
| 515 |
)
|
|
|
|
|
|
|
| 516 |
|
| 517 |
|
| 518 |
gender_selector = gr.Radio(
|