awalit / config.py
devcom33
Updated FastAPI code
0aee6bf
raw
history blame contribute delete
593 Bytes
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")