update

app.py

@@ -175,7 +175,7 @@ def generate_audio_data(snac_tokens, snacmodel, device=None):
     return audio_data
 
 
-# @torch.inference_mode()
+@torch.inference_mode()
 def run_AT_batch_stream(
                         audio_path,
                         stream_stride=4,
@@ -294,6 +294,10 @@ def run_AT_batch_stream(
     model.clear_kv_cache()
     return list_output
 
+    
+for chunk in run_AT_batch_stream('./data/samples/output1.wav'):
+    pass
+
 
 def process_audio(audio):
     filepath = audio