Spaces:

Ayushdavidkushwahaaaa
/

emotion_detection

Runtime error

Ayushdavidkushwahaaaa commited on Oct 2, 2024

Commit

3e23a55

verified ·

1 Parent(s): 22a38ef

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ warnings.filterwarnings('ignore')
 from transformers import pipeline
 import os
-MODEL_NAME = "openai/whisper-small"
 BATCH_SIZE = 8
 device = 0 if torch.cuda.is_available() else "cpu"

 from transformers import pipeline
 import os
+MODEL_NAME = "openai/whisper-medium"
 BATCH_SIZE = 8
 device = 0 if torch.cuda.is_available() else "cpu"