slplab
/

whisper-large_v2_test

Automatic Speech Recognition

Generated from Trainer

Inference Endpoints

Model card Files Files and versions Metrics Training metrics Community

slplab commited on Aug 14, 2023

Commit

2f68faf

•

1 Parent(s): 109cd63

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -18,7 +18,7 @@ class EndpointHandler:
         self.processor = WhisperProcessor.from_pretrained(path, language="korean", task='transcribe')
         #self.processor = AutoProcessor.from_pretrained(path)
         #self.pipe = pipeline("automatic-speech-recognition", model=model, tokenizer=processor.feature_extractor, feature_extractor=processor.feature_extractor)
-        self.feature_extractor = WhisperFeatureExtractor.from_pretrained(openai/whisper-large)
         # Move model to device

         self.processor = WhisperProcessor.from_pretrained(path, language="korean", task='transcribe')
         #self.processor = AutoProcessor.from_pretrained(path)
         #self.pipe = pipeline("automatic-speech-recognition", model=model, tokenizer=processor.feature_extractor, feature_extractor=processor.feature_extractor)
+        self.feature_extractor = WhisperFeatureExtractor.from_pretrained('openai/whisper-large')
         # Move model to device