daquanzhou commited on
Commit
8b6c408
·
1 Parent(s): 4b90430
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -233,7 +233,7 @@ class MagicMeController:
233
  self.id_embed_list = [Path(p).stem for p in id_embed_list]
234
 
235
  def run_once(self, prompt_text_box, negative_prompt_text_box, id_embed_dropdown, gaussian_slider, seed_text_box):
236
- category = "woman" if self.selected_id_embed in self.woman_id_embed_list else "man"
237
  prompt = f"a photo of embedding:{id_embed_dropdown} {category} " + prompt_text_box
238
  print("prompt:", prompt)
239
  print("negative_prompt_text_box:", negative_prompt_text_box)
@@ -536,7 +536,7 @@ def ui():
536
  """
537
  ### Quick Start
538
  1. Select desired `ID embedding`.
539
- 2. Provide `Prompt` and `Negative Prompt` for each model. Please use propoer pronoun for the character's gender.
540
  4. Click `Generate`, wait for ~5 min, and enjoy.
541
  """
542
  )
 
233
  self.id_embed_list = [Path(p).stem for p in id_embed_list]
234
 
235
  def run_once(self, prompt_text_box, negative_prompt_text_box, id_embed_dropdown, gaussian_slider, seed_text_box):
236
+ category = "woman" if id_embed_dropdown in self.woman_id_embed_list else "man"
237
  prompt = f"a photo of embedding:{id_embed_dropdown} {category} " + prompt_text_box
238
  print("prompt:", prompt)
239
  print("negative_prompt_text_box:", negative_prompt_text_box)
 
536
  """
537
  ### Quick Start
538
  1. Select desired `ID embedding`.
539
+ 2. Provide `Prompt` and `Negative Prompt`. Please use propoer pronoun for the character's gender.
540
  4. Click `Generate`, wait for ~5 min, and enjoy.
541
  """
542
  )