alexnasa commited on
Commit
586bceb
·
verified ·
1 Parent(s): a23408c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
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
- # 定义清空图像的函数,只返回四个 None
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")