Update app.py
Browse files
app.py
CHANGED
@@ -132,10 +132,11 @@ if config["model"]["use_dit_lora"]:
|
|
132 |
vae_skip_iter = None
|
133 |
attn_skip_iter = 0
|
134 |
|
135 |
-
|
136 |
def clear_images():
|
137 |
return [None, ]*num_inputs
|
138 |
|
|
|
139 |
def det_seg_img(image, label):
|
140 |
if isinstance(image, str):
|
141 |
image = Image.open(image).convert("RGB")
|
@@ -144,6 +145,7 @@ def det_seg_img(image, label):
|
|
144 |
ins, bbox = merge_instances(image, indices, instance_result_dict["instance_bboxes"], instance_result_dict["instance_images"])
|
145 |
return ins
|
146 |
|
|
|
147 |
def crop_face_img(image):
|
148 |
if isinstance(image, str):
|
149 |
image = Image.open(image).convert("RGB")
|
|
|
132 |
vae_skip_iter = None
|
133 |
attn_skip_iter = 0
|
134 |
|
135 |
+
|
136 |
def clear_images():
|
137 |
return [None, ]*num_inputs
|
138 |
|
139 |
+
@spaces.GPU()
|
140 |
def det_seg_img(image, label):
|
141 |
if isinstance(image, str):
|
142 |
image = Image.open(image).convert("RGB")
|
|
|
145 |
ins, bbox = merge_instances(image, indices, instance_result_dict["instance_bboxes"], instance_result_dict["instance_images"])
|
146 |
return ins
|
147 |
|
148 |
+
@spaces.GPU()
|
149 |
def crop_face_img(image):
|
150 |
if isinstance(image, str):
|
151 |
image = Image.open(image).convert("RGB")
|