JokerZhou commited on
Commit
4155ba3
·
verified ·
1 Parent(s): 70d8622

Update gradio_app_streaming.py

Browse files
Files changed (1) hide show
  1. gradio_app_streaming.py +2 -0
gradio_app_streaming.py CHANGED
@@ -15,6 +15,7 @@ import threading
15
  from datetime import datetime
16
  from collections import deque
17
  from loguru import logger
 
18
 
19
  from flash_head.inference import (
20
  get_pipeline,
@@ -83,6 +84,7 @@ def _save_chunk_audio_to_wav(audio_array, wav_path, sample_rate=16000):
83
  wav_file.writeframes(samples.tobytes())
84
  return wav_path
85
 
 
86
  def run_inference_streaming(
87
  ckpt_dir,
88
  wav2vec_dir,
 
15
  from datetime import datetime
16
  from collections import deque
17
  from loguru import logger
18
+ import spaces
19
 
20
  from flash_head.inference import (
21
  get_pipeline,
 
84
  wav_file.writeframes(samples.tobytes())
85
  return wav_path
86
 
87
+ @spaces.GPU
88
  def run_inference_streaming(
89
  ckpt_dir,
90
  wav2vec_dir,