Browse Source

add save src face

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

+ 4 - 4
api/main.py

@@ -94,7 +94,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= im.convert("RGB")
+    im.save('../../face_swap/tmp_face/'+name_hash+'.jpg')
     x = threading.Thread(target=gen_avatar, args=(name_hash,req.imgurl))
     x.start()
     return {'msg':'人物生成中,請稍候'}
@@ -178,9 +180,7 @@ def gen_video(name_hash,name,text_content, image_urls,avatar,client_id):
 
 
 def gen_avatar(name_hash, imgurl):
-    im = Image.open(requests.get(imgurl, stream=True).raw)
-    im= im.convert("RGB")
-    im.save('../../face_swap/tmp_face/'+name_hash+'.jpg')
+    
     
     while True:
         directory= os.listdir('../../face_swap/tmp_face/')