FaceSwapping/models/inswapper/__init__.py

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"  # Thay URL thật ở đây

# Tạo thư mục nếu chưa tồn tại
os.makedirs(PATH_FOLDER, exist_ok=True)

# Kiểm tra nếu không có file .onnx thì tải về
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}")