|
|
@@ -59,6 +59,7 @@ tmp_img_sub_folder = 'tmp_img/'
|
|
|
img_upload_folder = '/var/www/html/'+tmp_img_sub_folder
|
|
|
video_dest = '/var/www/html/'+video_sub_folder
|
|
|
avatar_dest = '/var/www/html/'+avatar_sub_folder
|
|
|
+tmp_src_face ='../../face_swap/tmp_face/'
|
|
|
|
|
|
class swap_req(BaseModel):
|
|
|
imgurl: str
|
|
|
@@ -94,9 +95,9 @@ async def swapFace(req:swap_req):
|
|
|
except:
|
|
|
return {'msg':"無法辨別圖片網址"+req.imgurl}
|
|
|
name_hash = str(time.time()).replace('.','')
|
|
|
- im = Image.open(requests.get(imgurl, stream=True).raw)
|
|
|
+ im = Image.open(requests.get(req.imgurl, stream=True).raw)
|
|
|
im= im.convert("RGB")
|
|
|
- im.save('../../face_swap/tmp_face/'+name_hash+'.jpg')
|
|
|
+ im.save(tmp_src_face+name_hash+'.jpg')
|
|
|
x = threading.Thread(target=gen_avatar, args=(name_hash))
|
|
|
x.start()
|
|
|
return {'msg':'人物生成中，請稍候'}
|
