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

c3cd4b7

·

1 Parent(s): 57b2de0

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -2,7 +2,7 @@ from typing import Dict, List, Any
 from transformers import pipeline
 class EndpointHandler():
-    def __init__(self, path=""):
         self.pipeline = pipeline(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!

 from transformers import pipeline
 class EndpointHandler():
+    def __init__(self, path="Evan-Lin/whisper-large-v1-tw"):
         self.pipeline = pipeline(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!