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

f0f40a8

·

1 Parent(s): c22159d

Update ffmpeg_handler.py

Files changed (1) hide show

ffmpeg_handler.py +1 -1

ffmpeg_handler.py CHANGED Viewed

@@ -32,7 +32,7 @@ class EndpointHandler:
         audio_tensor= torch.from_numpy(audio_nparray)
-        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(audio_tensor)

         audio_tensor= torch.from_numpy(audio_nparray)
+        transcribe = self.pipe
         transcribe.model.config.forced_decoder_ids = transcribe.tokenizer.get_decoder_prompt_ids(language="ko", task="transcribe")
         result = transcribe(audio_tensor)