Spaces:
Runtime error
Runtime error
File size: 1,994 Bytes
c19ca42 |
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 45 46 47 48 49 50 51 |
import os
from modules.shared import log, opts
from modules import devices
insightface_app = None
instightface_mp = None
def get_app(mp_name):
global insightface_app, instightface_mp # pylint: disable=global-statement
from installer import installed, install
packages = [
('insightface', 'insightface'),
('git+https://github.com/tencent-ailab/IP-Adapter.git', 'ip_adapter'),
]
for pkg in packages:
if not installed(pkg[1], reload=False, quiet=True):
install(pkg[0], pkg[1], ignore=False)
if insightface_app is None or mp_name != instightface_mp:
from insightface.app import FaceAnalysis
import huggingface_hub as hf
import zipfile
log.debug(f"InsightFace: mp={mp_name} provider={devices.onnx}")
root_dir = os.path.join(opts.diffusers_dir, 'models--vladmandic--insightface-faceanalysis')
local_dir = os.path.join(root_dir, 'models')
extract_dir = os.path.join(local_dir, mp_name)
model_path = os.path.join(local_dir, f'{mp_name}.zip')
if not os.path.exists(model_path):
model_path = hf.hf_hub_download(
repo_id='vladmandic/insightface-faceanalysis',
filename=f'{mp_name}.zip',
local_dir_use_symlinks=False,
cache_dir=opts.diffusers_dir,
local_dir=local_dir
)
if not os.path.exists(extract_dir):
log.debug(f'InsightFace extract: folder="{extract_dir}"')
os.makedirs(extract_dir)
with zipfile.ZipFile(model_path) as zf:
zf.extractall(local_dir)
kwargs = {
'root': root_dir,
'download': False,
'download_zip': False,
}
insightface_app = FaceAnalysis(name=mp_name, providers=devices.onnx, **kwargs)
instightface_mp = mp_name
insightface_app.prepare(ctx_id=0, det_thresh=0.5, det_size=(640, 640))
return insightface_app
|