mimi codes shape warning

app.py

@@ -26,8 +26,10 @@ def compute_codes(wav):
             for offset in tqdm(range(0, wav.shape[-1], frame_size), desc="computing Codes"):
                 frame = wav[:, :, offset: offset + frame_size]
                 codes = mimi.encode(frame)
-                assert codes.shape[-1] == 1, codes.shape
-                all_codes.append(codes)
+                if codes.shape[-1] == 1:
+                    all_codes.append(codes)
+                else:
+                    print(f"Warning: Empty codes for frame at offset {offset}")
     return all_codes
 
 @spaces.GPU
@@ -100,8 +102,8 @@ def truncate_audio(wav, max_length):
 
 def process_audio(audio, instream):
     log_out = ""
-    outwav = torch.randn(1, 1, 24000 * 10)
-    stream = torch.randn(1, 1, 24000 * 10)
+    outwav = torch.randn(1, 1, 24000 * 2)
+    stream = torch.randn(1, 1, 24000 * 2)
 
 
     print("Audio recieved")