config.py

# Purpose: ArtifactNet HF Spaces demo — constants (HF Hub paths)

"""HF Spaces (CPU / onnxruntime) 전용 설정.

가중치는 HF Hub 의 intrect/artifactnet-models 에서 .onnx 만 다운로드한다
(.pt 는 private).
"""

# HF Hub model repo
HF_MODEL_REPO = "intrect/artifactnet-models"
UNET_ONNX_FILENAME = "production-onnx/unet_codec4.onnx"
CNN_ONNX_FILENAME  = "production-onnx/cnn_v94.onnx"

# Audio constants
SR = 44100
MAX_DURATION_SEC = 300
CHUNK_SEC = 4.0
CHUNK_SAMPLES = int(CHUNK_SEC * SR)

# STFT
N_FFT = 2048
HOP_LENGTH = 512

# CNN batch (CPU 에선 2 vCPU 기준 너무 크면 latency 증가)
BATCH_SIZE = 4

# 60 초 미만 파일은 "Too Short" 경고
MIN_CONFIDENT_DURATION = 60