MoYoYoTech
/

Translator

Model card Files Files and versions

liumaolin commited on Apr 23

Commit

80839d5

·

1 Parent(s): 0c9fcfc

Disable FunASR pbar.

Files changed (1) hide show

transcribe/helpers/funasr.py +1 -1

transcribe/helpers/funasr.py CHANGED Viewed

@@ -30,7 +30,7 @@ class FunASR:
         audio_frames = np.frombuffer(audio_buffer, dtype=np.float32)
         # sf.write(f'{config.ASSERT_DIR}/{time.time()}.wav', audio_frames, samplerate=16000)
         try:
-            output = self.model.generate(input=audio_frames)
             return output
         except Exception as e:
             logger.error(e)

         audio_frames = np.frombuffer(audio_buffer, dtype=np.float32)
         # sf.write(f'{config.ASSERT_DIR}/{time.time()}.wav', audio_frames, samplerate=16000)
         try:
+            output = self.model.generate(input=audio_frames, disable_pbar=True)
             return output
         except Exception as e:
             logger.error(e)