Significant commited on
Commit
eb0a568
·
verified ·
1 Parent(s): 89980a1

Update app.py

Browse files

dropdown prompt

Files changed (1) hide show
  1. app.py +11 -2
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(