Update app.py

app.py

@@ -167,6 +167,12 @@ def generate_audio(script_output, voice1, voice2, num_hosts, temperature, top_p,
                 )
             
             code_list = parse_output(generated_ids)
+            
+            # Ensure the code list is long enough (at least 7 elements)
+            min_length = 7
+            if len(code_list) < min_length:
+                code_list = code_list + [0] * (min_length - len(code_list))
+            
             # Use the correct method for audio generation based on your SNAC model
             codes_tensor = torch.tensor(code_list).unsqueeze(0).to(device)
             paragraph_audio = snac_model(codes_tensor)