Browse Source

add save src face

ming 3 years ago
parent
commit
4e880c973c
1 changed files with 2 additions and 2 deletions
  1. 2 2
      api/main.py

+ 2 - 2
api/main.py

@@ -180,8 +180,8 @@ def gen_video(name_hash,name,text_content, image_urls,avatar,client_id):
 def gen_avatar(name_hash, imgurl):
     while True:
 
-        directory= os.listdir('../../face_swap/tmp_avatar/') 
-        if len(directory) == 0 and not os.path.isdir('../../face_swap/queueFlag'):
+        directory= os.listdir('../../face_swap/tmp_face/') 
+        if len(directory) == 0 and os.path.isdir('../../face_swap/queueFlag'):
             os.mkdir('../../face_swap/queueFlag')
             c = rpyc.connect("localhost", 8868)
             c._config['sync_request_timeout'] = None