add: PIPNet make.sh when started

app.py

@@ -176,6 +176,7 @@ def make_pipnet():
     ]
     for cmd in cmds:
         os.system(cmd)
+    print('[PIPNet] nms .o file built successfully.')
 
 
 def swap_image(
@@ -447,7 +448,6 @@ def swap_video_gr(img1, target_path, use_gpu=True, frames=9999999):
 
 if __name__ == "__main__":
     use_gpu = torch.cuda.is_available()
-    make_pipnet()
 
     with gr.Blocks() as demo:
         gr.Markdown("SuperSwap")

third_party/PIPNet/lib/tools.py

@@ -1,6 +1,8 @@
 import cv2
 import sys
 
+from app import make_pipnet
+make_pipnet()
 
 from math import floor
 from third_party.PIPNet.FaceBoxesV2.faceboxes_detector import *