liumaolin
commited on
Commit
·
4c23586
1
Parent(s):
c5cc64b
Disable FunASR pbar in Warmup.
Browse files
transcribe/helpers/funasr.py
CHANGED
|
@@ -24,7 +24,7 @@ class FunASR:
|
|
| 24 |
def warmup(self, warmup_steps=1):
|
| 25 |
warmup_soundfile = f"{config.ASSERT_DIR}/jfk.flac"
|
| 26 |
for _ in range(warmup_steps):
|
| 27 |
-
self.model.generate(input=warmup_soundfile)
|
| 28 |
|
| 29 |
def transcribe(self, audio_buffer: bytes, language):
|
| 30 |
audio_frames = np.frombuffer(audio_buffer, dtype=np.float32)
|
|
|
|
| 24 |
def warmup(self, warmup_steps=1):
|
| 25 |
warmup_soundfile = f"{config.ASSERT_DIR}/jfk.flac"
|
| 26 |
for _ in range(warmup_steps):
|
| 27 |
+
self.model.generate(input=warmup_soundfile, disable_pbar=True)
|
| 28 |
|
| 29 |
def transcribe(self, audio_buffer: bytes, language):
|
| 30 |
audio_frames = np.frombuffer(audio_buffer, dtype=np.float32)
|