Spaces:
Running
Running
File size: 867 Bytes
79230dc |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
WEIGHT_URLS = {
'buffalo_l':
'https://github.com/justld/dofaker/releases/download/v0.1/buffalo_l.zip',
'buffalo_s':
'https://github.com/justld/dofaker/releases/download/v0.1/buffalo_s.zip',
'buffalo_sc':
'https://github.com/justld/dofaker/releases/download/v0.1/buffalo_sc.zip',
'inswapper':
'https://github.com/justld/dofaker/releases/download/v0.1/inswapper_128.onnx',
'gfpgan':
'https://github.com/justld/dofaker/releases/download/v0.1/GFPGANv1.3.onnx',
'bsrgan':
'https://github.com/justld/dofaker/releases/download/v0.1/bsrgan_4.onnx',
'openpose_body':
'https://github.com/justld/dofaker/releases/download/v0.1/openpose_body.onnx',
'pose_transfer':
'https://github.com/justld/dofaker/releases/download/v0.1/pose_transfer.onnx',
}
def get_model_url(model_name):
return WEIGHT_URLS[model_name]
|