Spaces:

whisper-event
/

whisper-demo

Runtime error

reach-vb HF Staff commited on Dec 4, 2022

Commit

cc3c457

1 Parent(s): 495c693

Update app.py (#5)

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,7 +7,7 @@ from huggingface_hub import model_info
 MODEL_NAME = "openai/whisper-small" #this always needs to stay in line 8 :D sorry for the hackiness
-device = "cuda" if torch.cuda.is_available() else "cpu"
 pipe = pipeline(
     task="automatic-speech-recognition",

 MODEL_NAME = "openai/whisper-small" #this always needs to stay in line 8 :D sorry for the hackiness
+device = 0 if torch.cuda.is_available() else "cpu"
 pipe = pipeline(
     task="automatic-speech-recognition",