Spaces:
Configuration error
Configuration error
File size: 1,106 Bytes
2c050c4 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 |
#!/usr/bin/python
# -*- encoding: utf-8 -*-
from io import BytesIO
import time
import base64
import json
import requests
key = "-fd9YqPnrLnmugQGAhQoimCkQd0t8N8L"
secret = "0GLyRIHDnrjKSlDuflLPO8a6U32hyDUy"
def encode(image: 'PIL.Image') -> str:
with BytesIO() as output_buf:
image.save(output_buf, format='PNG')
return base64.b64encode(output_buf.getvalue()).decode('utf-8')
def beautify(image: 'PIL.Image') -> str:
data = {
'api_key': key,
'api_secret': secret,
'image_base64': encode(image),
}
resp = requests.post(beautify.url, data=data)
return resp.json()['result']
def rank(image: 'PIL.Image') -> int:
data = {
'api_key': key,
'api_secret': secret,
'image_base64': encode(image),
'return_attributes': 'beauty',
}
resp = requests.post(rank.url, data=data)
scores = resp.json()['faces'][0]['attributes']['beauty']
return max(scores.values())
beautify.url = 'https://api-cn.faceplusplus.com/facepp/v2/beautify'
rank.url = 'https://api-cn.faceplusplus.com/facepp/v3/detect'
|