model_manager.py

# =====================================================
# Apckeyl_RealESRGAN
# Version 6.1
# model_manager.py
# =====================================================

from huggingface_hub import hf_hub_download

from config import (
    HF_REPO,
    HF_FILENAME,
    MODEL_PATH,
)


def get_model_path():

    if MODEL_PATH.exists():

        print(f"✓ Используется локальная модель\n{MODEL_PATH}")

        return str(MODEL_PATH)

    print(f"Скачивание модели {HF_FILENAME}...")

    hf_hub_download(
        repo_id=HF_REPO,
        filename=HF_FILENAME,
        local_dir=MODEL_PATH.parent,
    )

    print(f"✓ {HF_FILENAME} успешно скачана")

    return str(MODEL_PATH)