speech-to-speech-translation

Sleeping

arshsin commited on Feb 15

Commit

83d9770

•

1 Parent(s): a7cca5d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,7 +9,7 @@ from transformers import pipeline
 device = "cuda:0" if torch.cuda.is_available() else "cpu"
 # load speech translation checkpoint
-asr_pipe = pipeline("automatic-speech-recognition", model="openai/whisper-base", device=device)

 device = "cuda:0" if torch.cuda.is_available() else "cpu"
 # load speech translation checkpoint
+asr_pipe = pipeline("automatic-speech-recognition", model="openai/whisper-tiny", device=device)