File size: 470 Bytes
967b24c
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
import os
from dotenv import load_dotenv
from faster_whisper import WhisperModel

load_dotenv()
##Cau hinh cho Whisper

# WHISPER_MODEL_SIZE = 'large-v3'
# WHISPER_DEVICE = 'cuda'
# WHISPER_COMPUTE_TYPE = 'float16'
# WHISPER_BEAM_SIZE = 8
# WHISPER_USE_VAD = True

WHISPER_MODEL_SIZE = 'medium'
WHISPER_DEVICE = 'cpu'
WHISPER_COMPUTE_TYPE = 'int8'
WHISPER_BEAM_SIZE = 8
WHISPER_USE_VAD = True

## OPen ai key
OPENAI_API_KEY = os.getenv('OPEN_AI_KEY')