Spaces:

DarwinAnim8or
/

Whisper-Demo

Sleeping

DarwinAnim8or commited on Aug 10, 2023

Commit

713bc64

1 Parent(s): 253867d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,7 +4,7 @@ from transformers import pipeline
 # Load the model and processor
 model_id = "openai/whisper-medium"
-device = 0 if torch.cuda.is_available() else "cpu"
 BATCH_SIZE = 8
 pipe = pipeline(
     task="automatic-speech-recognition",

 # Load the model and processor
 model_id = "openai/whisper-medium"
+device = "cpu"
 BATCH_SIZE = 8
 pipe = pipeline(
     task="automatic-speech-recognition",