|
|
@@ -98,7 +98,8 @@ async def swapFace(req:swap_req):
|
|
|
im = Image.open(requests.get(req.imgurl, stream=True).raw)
|
|
|
im= im.convert("RGB")
|
|
|
im.save(tmp_src_face+name_hash+'.jpg')
|
|
|
- x = threading.Thread(target=gen_avatar, args=(str(name_hash)))
|
|
|
+
|
|
|
+ x = threading.Thread(target=gen_avatar, args=(name_hash,req.imgurl))
|
|
|
x.start()
|
|
|
return {'msg':'人物生成中，請稍候'}
|
|
|
|
|
|
@@ -180,7 +181,7 @@ def gen_video(name_hash,name,text_content, image_urls,avatar,client_id):
|
|
|
os.remove(tmp_video_dir+name_hash+'.mp4')
|
|
|
|
|
|
|
|
|
-def gen_avatar(name_hash):
|
|
|
+def gen_avatar(name_hash,imgurl):
|
|
|
|
|
|
c = rpyc.connect("localhost", 8868)
|
|
|
c._config['sync_request_timeout'] = None
|
