S-Fry
/

large

Automatic Speech Recognition

hf-asr-leaderboard

Inference Endpoints

Model card Files Files and versions Community

S-Fry commited on Apr 27, 2023

Commit

6452c1e

•

1 Parent(s): afacbcf

Update handler.py

Files changed (1) hide show

handler.py +2 -2

handler.py CHANGED Viewed

@@ -7,8 +7,8 @@ from transformers.pipelines.audio_utils import ffmpeg_read
 SAMPLE_RATE=16000
 class EndpointHandler():
     def __init__(self, path=""):
-        self.device = "cuda:0" if torch.cuda.is_available() else "cpu"
-        self.pipe = pipeline(
           "automatic-speech-recognition",
           model="openai/whisper-large",
           chunk_length_s=30,

 SAMPLE_RATE=16000
 class EndpointHandler():
     def __init__(self, path=""):
+        device = "cuda:0" if torch.cuda.is_available() else "cpu"
+        pipe = pipeline(
           "automatic-speech-recognition",
           model="openai/whisper-large",
           chunk_length_s=30,