ming преди 3 години
родител
ревизия
2341327e7d
променени са 1 файла, в които са добавени 6 реда и са изтрити 1 реда
  1. 6 1
      api/util/swap_face.py

+ 6 - 1
api/util/swap_face.py

@@ -21,11 +21,16 @@ class swap_face():
         x = threading.Thread(target=self.runthreadswap, args=(src_img,sv_path))
         x.start()
         time.sleep(20)
+        load_time = 0
         while True:
             print('waiting...')
             if os.path.exists(sv_path):
                 break
-            time.sleep(5)
+            time.sleep(10)
+            load_time+=10
+            if load_time > 30:
+                break
+        
             print('waiting...')
         self.notify_group('人物生成成功,在:www.choozmo.com:8168/swap_save/'+name_hash+'.avi')
         return {'msg':'生成中,成果會在line中展示'}