denoiser_model.py

import torch
from denoiser import pretrained

device = "cuda" if torch.cuda.is_available() else "cpu"

# Load DNS64 pretrained model (auto-downloads if not cached)
denoiser_model = pretrained.dns64().to(device)
denoiser_model.eval()