File size: 372 Bytes
5f0a430
85c98be
5f0a430
85c98be
5f0a430
 
85c98be
5f0a430
85c98be
5f0a430
 
 
85c98be
5f0a430
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
import os
import psutil

WHISPER_MODEL_NAME = "Systran/faster-whisper-tiny"
WHISPER_DEVICE = "cpu"
WHISPER_COMPUTE_TYPE = "int8"

PYANNOTE_AUTH_TOKEN = os.getenv("HUGGINGFACE_API_KEY")

SUMMARIZER_MODEL = "facebook/bart-large-cnn"
SUMMARIZER_MAX_LENGTH = 150
SUMMARIZER_MIN_LENGTH = 50

SPACY_MODEL = "en_core_web_sm"
CPU_THREADS = max(1, psutil.cpu_count(logical=False))