gpt-omni commited on
Commit
aab4898
1 Parent(s): 56fd6ef
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -173,6 +173,7 @@ def generate_audio_data(snac_tokens, snacmodel, device=None):
173
  return audio_data
174
 
175
 
 
176
  @torch.inference_mode()
177
  def run_AT_batch_stream(
178
  audio_path,
 
173
  return audio_data
174
 
175
 
176
+ @spaces.GPU
177
  @torch.inference_mode()
178
  def run_AT_batch_stream(
179
  audio_path,