Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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がロードされている場合は再利用
|