Update app.py

app.py

@@ -13,12 +13,16 @@ from transformers import pipeline, Wav2Vec2ProcessorWithLM
 from pyannote.audio import Pipeline
 import whisperx
 
-from utils import split, create_fig, color_map, thresholds
+from utils import split, create_fig
 from utils import speech_to_text as stt
 
 os.environ["TOKENIZERS_PARALLELISM"] = "false"
 device = 0 if torch.cuda.is_available() else -1
 
+# display if the sentiment value is above these thresholds
+thresholds = {"joy": 0.99,"anger": 0.95,"surprise": 0.95,"sadness": 0.98,"fear": 0.95,"love": 0.99,}
+
+color_map = {"joy": "green","anger": "red","surprise": "yellow","sadness": "blue","fear": "orange","love": "purple",}
 
 # Audio components
 whisper_device = "cuda" if torch.cuda.is_available() else "cpu"