Jelajahi Sumber

fix gen avatar

ming 3 tahun lalu
induk
melakukan
2341327e7d
1 mengubah file dengan 6 tambahan dan 1 penghapusan
  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中展示'}