Explorar el Código

add save src face

ming hace 3 años
padre
commit
4e880c973c
Se han modificado 1 ficheros con 2 adiciones y 2 borrados
  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