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

c22159d

•

1 Parent(s): 099a81c

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -26,7 +26,7 @@ class EndpointHandler:
     def __call__(self, data: Any) -> List[Dict[str, str]]:
         print('==========NEW PROCESS=========')
-        transcribe = pipeline(task="automatic-speech-recognition", model="vasista22/whisper-kannada-tiny")
         transcribe.model.config.forced_decoder_ids = transcribe.tokenizer.get_decoder_prompt_ids(language="ko", task="transcribe")
         result = transcribe(data['inputs'])

     def __call__(self, data: Any) -> List[Dict[str, str]]:
         print('==========NEW PROCESS=========')
+        transcribe = self.pipe
         transcribe.model.config.forced_decoder_ids = transcribe.tokenizer.get_decoder_prompt_ids(language="ko", task="transcribe")
         result = transcribe(data['inputs'])