import psutil | |
import os | |
# Whisper settings | |
WHISPER_MODEL_NAME = "tiny" | |
WHISPER_DEVICE = "cpu" | |
WHISPER_COMPUTE_TYPE = "int8" | |
# Use token from Hugging Face Space secrets | |
PYANNOTE_AUTH_TOKEN = os.environ.get("HUGGINGFACE_API_KEY") | |
# Summarization settings | |
SUMMARIZER_MODEL = "facebook/bart-large-cnn" | |
SUMMARIZER_MAX_LENGTH = 150 | |
SUMMARIZER_MIN_LENGTH = 50 | |
# NLP settings | |
SPACY_MODEL = "en_core_web_sm" | |
CPU_THREADS = max(1, psutil.cpu_count(logical=False)) | |
# Check that token is set | |
if not PYANNOTE_AUTH_TOKEN: | |
raise ValueError("HUGGINGFACE_API_KEY is not set in environment variables") | |