| from app.config import settings
|
| import os
|
| import urllib.request
|
|
|
| PATH_FOLDER = os.path.join(settings.DIR_ROOT, "FaceSwapping", "models", "inswapper")
|
| ONNX_FILENAME = "inswapper_128.onnx"
|
|
|
| ONNX_PATH = os.path.join(PATH_FOLDER, ONNX_FILENAME)
|
| ONNX_URL = "https://huggingface.co/duyv/MC-AI/resolve/main/FaceSwapping/models/inswapper/inswapper_128.onnx"
|
|
|
|
|
| os.makedirs(PATH_FOLDER, exist_ok=True)
|
|
|
|
|
| if not os.path.exists(ONNX_PATH):
|
| print(f"Tải {ONNX_FILENAME} từ {ONNX_URL}...")
|
| urllib.request.urlretrieve(ONNX_URL, ONNX_PATH)
|
| print("Tải xong.")
|
| else:
|
| print(f"Đã có file {ONNX_FILENAME} tại {ONNX_PATH}")
|
|
|
