Spaces:

hamza666
/

Speech-to-speech-translation

Sleeping

hamza666 commited on Sep 4

Commit

7264692

•

1 Parent(s): c4d170b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,11 +1,10 @@
-import torch
 from transformers import pipeline
 from datasets import load_dataset
 from datasets import Audio
 from transformers import SpeechT5Processor, SpeechT5ForTextToSpeech, SpeechT5HifiGan
-device = "cuda:0" if torch.cuda.is_available() else "cpu"
 pipe = pipeline(
     "automatic-speech-recognition", model="openai/whisper-base", device=device
 )

 from transformers import pipeline
 from datasets import load_dataset
 from datasets import Audio
 from transformers import SpeechT5Processor, SpeechT5ForTextToSpeech, SpeechT5HifiGan
+device = "cpu"
 pipe = pipeline(
     "automatic-speech-recognition", model="openai/whisper-base", device=device
 )