gavinyuan commited on
Commit
51779c3
1 Parent(s): b9be4e6

add: PIPNet make.sh when started

Browse files
Files changed (1) hide show
  1. app.py +13 -0
app.py CHANGED
@@ -167,6 +167,17 @@ else:
167
  print(f'[demo] model loaded from {pt_path}')
168
 
169
 
 
 
 
 
 
 
 
 
 
 
 
170
  def swap_image(
171
  source_image,
172
  target_path,
@@ -436,6 +447,8 @@ def swap_video_gr(img1, target_path, use_gpu=True, frames=9999999):
436
 
437
  if __name__ == "__main__":
438
  use_gpu = torch.cuda.is_available()
 
 
439
  with gr.Blocks() as demo:
440
  gr.Markdown("SuperSwap")
441
 
 
167
  print(f'[demo] model loaded from {pt_path}')
168
 
169
 
170
+ def make_pipnet():
171
+ cmds = [
172
+ "cd ./third_party/PIPNet/FaceBoxesV2/utils/nms/"
173
+ "chmod +x make.sh"
174
+ "bash ./make.sh"
175
+ "cd -"
176
+ ]
177
+ for cmd in cmds:
178
+ os.system(cmd)
179
+
180
+
181
  def swap_image(
182
  source_image,
183
  target_path,
 
447
 
448
  if __name__ == "__main__":
449
  use_gpu = torch.cuda.is_available()
450
+ make_pipnet()
451
+
452
  with gr.Blocks() as demo:
453
  gr.Markdown("SuperSwap")
454