tori29umai commited on
Commit
4f189ae
·
verified ·
1 Parent(s): c6c6880

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -0
app.py CHANGED
@@ -67,6 +67,17 @@ class Img2Img:
67
  self.bg_removed_image = None
68
  self.pipe = None
69
  self.current_lora_model = None
 
 
 
 
 
 
 
 
 
 
 
70
 
71
  def load_model(self, lora_model):
72
  # 既に正しいpipeがロードされている場合は再利用
 
67
  self.bg_removed_image = None
68
  self.pipe = None
69
  self.current_lora_model = None
70
+
71
+ def process_prompt_analysis(self, input_image_path):
72
+ if self.tagger_model is None:
73
+ self.tagger_model = modelLoad(tagger_dir)
74
+ tags = analysis(input_image_path, tagger_dir, self.tagger_model)
75
+ prompt = remove_color(tags)
76
+ execute_tags = ["realistic", "nose", "asian"]
77
+ prompt = execute_prompt(execute_tags, prompt)
78
+ prompt = remove_duplicates(prompt)
79
+ return prompt
80
+
81
 
82
  def load_model(self, lora_model):
83
  # 既に正しいpipeがロードされている場合は再利用