cathyi commited on
Commit
8f4ba23
·
1 Parent(s): 3082d5f

update handler

Browse files
Files changed (1) hide show
  1. 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
  """