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

86435c4

•

1 Parent(s): 2f68faf

Update handler.py

Files changed (1) hide show

handler.py +2 -2

handler.py CHANGED Viewed

@@ -12,13 +12,13 @@ import io
 class EndpointHandler:
     def __init__(self, path=""):
-        self.tokenizer = WhisperTokenizer.from_pretrained(path, language="korean", task='transcribe')
         self.model = WhisperForConditionalGeneration.from_pretrained(path)
         #self.tokenizer = WhisperTokenizer.from_pretrained(path)
         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

 class EndpointHandler:
     def __init__(self, path=""):
+        self.tokenizer = WhisperTokenizer.from_pretrained('openai/whisper-large', language="korean", task='transcribe')
         self.model = WhisperForConditionalGeneration.from_pretrained(path)
         #self.tokenizer = WhisperTokenizer.from_pretrained(path)
         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(path)
         # Move model to device