-
Notifications
You must be signed in to change notification settings - Fork 5
/
Copy pathconfig.yaml
37 lines (31 loc) · 1.46 KB
/
config.yaml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
debug: true # Shared debug setting across all modules
# Application Configuration
flask:
app_name: Sentiment Analysis API
host: 0.0.0.0
port: 8001
debug: true
# Audio Configuration
# audio:
# Transcription Configuration
transcription:
default_model: "whisper" # Specify the default transcription model (e.g., whisper, another_model)
whisper: # Whisper-specific configuration
model_size: "base" # Choose between tiny, base, small, medium, large
device: 'cpu' # `cpu` for CPU, or `cuda` GPU device
chunk_length_s: 30
# another_model: # Placeholder for another transcription model's configuration
# api_key: "your_api_key"
# endpoint: "https://api.example.com/transcribe"
# Sentiment Analysis Configuration
sentiment_analysis:
default_model: "bertweet" # Specify the default sentiment analysis model (e.g., bertweet, another_model)
bertweet: # Vader-specific configuration
model_name: "finiteautomata/bertweet-base-sentiment-analysis"
device: 'cpu' # `cpu` for CPU, or `cuda` GPU device
# another_model: # Placeholder for another sentiment analysis model's configuration
# api_key: "your_api_key"
# endpoint: "https://api.example.com/sentiment"
# AudioTranscriptionSentimentPipeline Configuration
audio_transcription_sentiment_pipeline:
remove_audio: false # Specify whether to remove audio files after processing