update handler
Browse files- handler.py +0 -1
handler.py
CHANGED
@@ -28,7 +28,6 @@ class EndpointHandler():
|
|
28 |
# self.pipeline = pipeline(task= "automatic-speech-recognition", model=model, tokenizer=tokenizer, feature_extractor=feature_extractor)
|
29 |
self.pipeline.model.config.forced_decoder_ids = self.pipeline.tokenizer.get_decoder_prompt_ids(language="Chinese", task="transcribe")
|
30 |
self.pipeline.model.generation_config.forced_decoder_ids = self.pipeline.model.config.forced_decoder_ids # just to be sure!
|
31 |
-
# self.pipeline = pipeline(task= "automatic-speech-recognition", model=self.model
|
32 |
|
33 |
def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
|
34 |
"""
|
|
|
28 |
# self.pipeline = pipeline(task= "automatic-speech-recognition", model=model, tokenizer=tokenizer, feature_extractor=feature_extractor)
|
29 |
self.pipeline.model.config.forced_decoder_ids = self.pipeline.tokenizer.get_decoder_prompt_ids(language="Chinese", task="transcribe")
|
30 |
self.pipeline.model.generation_config.forced_decoder_ids = self.pipeline.model.config.forced_decoder_ids # just to be sure!
|
|
|
31 |
|
32 |
def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
|
33 |
"""
|