Evan-Lin
/

whisper-large-v1-tw

Automatic Speech Recognition

hf-asr-leaderboard

Inference Endpoints

Model card Files Files and versions Community

Evan-Lin commited on May 8, 2023

Commit

7f3abba

•

1 Parent(s): 8d88f31

Update handler.py

Files changed (1) hide show

handler.py +2 -2

handler.py CHANGED Viewed

@@ -3,8 +3,8 @@ from transformers import pipeline
 class EndpointHandler():
     def __init__(self, path=""):
-        self.pipeline = pipeline("automatic-speech-recognition", model=path)
-        self.pipeline.model.config.forced_decoder_ids = self.pipeline.model.processor.get_decoder_prompt_ids(language="Chinese", task="transcribe")
         self.pipeline.model.generation_config.forced_decoder_ids = self.pipeline.model.config.forced_decoder_ids  # just to be sure!
     def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:

 class EndpointHandler():
     def __init__(self, path=""):
+        self.pipeline = pipeline(task= "automatic-speech-recognition", model=path)
+        self.pipeline.model.config.forced_decoder_ids = self.pipeline.tokenizer.get_decoder_prompt_ids(language="Chinese", task="transcribe")
         self.pipeline.model.generation_config.forced_decoder_ids = self.pipeline.model.config.forced_decoder_ids  # just to be sure!
     def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]: