Hugging Face's logo

Spaces:
zongxiao
/
speech-to-speech
Paused

App Files Community
zongxiao commited on
Commit
374bd91
·
1 Parent(s): 479a3c0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ from transformers import pipeline
5
  from transformers import BarkModel
6
  from transformers import AutoProcessor
7
 
8
- device="cpu"
9
 
10
  pipe = pipeline(
11
  "automatic-speech-recognition", model="openai/whisper-large-v2", device=device
 
5
  from transformers import BarkModel
6
  from transformers import AutoProcessor
7
 
8
+ device = "cuda:0" if torch.cuda.is_available() else "cpu"
9
 
10
  pipe = pipeline(
11
  "automatic-speech-recognition", model="openai/whisper-large-v2", device=device