pitchflower_app/config.py

import os, torch

TARGET_SR = 24_000
WORLD_FRAME_PERIOD_MS = 32.0 / 3.0  # hop 256 @ 24k
DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
DEFAULT_CHECKPOINT_DIR = os.getenv("PITCHFLOWER_CHECKPOINT", "checkpoint/")
VOCOS_ID = os.getenv("VOCOS_ID", "charactr/vocos-mel-24khz")
MAX_SECONDS = float(os.getenv("MAX_SECONDS", "20"))

# Canvas
CANVAS_W, CANVAS_H = 1000, 320
MARG_L, MARG_R, MARG_T, MARG_B = 60, 20, 20, 40