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

d3d7605

·

1 Parent(s): 07e34c4

Update handler.py

Files changed (1) hide show

handler.py +2 -2

handler.py CHANGED Viewed

@@ -6,8 +6,8 @@ from datasets import load_dataset
 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,

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