Update kokoro.py

kokoro.py

@@ -131,19 +131,14 @@ def forward(model, tokens, ref_s, speed):
     asr = t_en @ pred_aln_trg.unsqueeze(0).to(device)
     return model.decoder(asr, F0_pred, N_pred, ref_s[:, :128]).squeeze().cpu().numpy()
 
-def generate(model, text, voicepack, speed=1, phonetics=None):
-    if phonetics is not None:
-        ps = phonetics
-    else:
-        ps = phonemize(text)
-    
+def generate(model, text, voicepack, speed=1, ps=None):
+    ps = ps or phonemize(text)
     tokens = tokenize(ps)
     if not tokens:
         return None
     elif len(tokens) > 510:
         tokens = tokens[:510]
         print('Truncated to 510 tokens')
-    
     ref_s = voicepack[len(tokens)]
     out = forward(model, tokens, ref_s, speed)
     ps = ''.join(next(k for k, v in VOCAB.items() if i == v) for i in tokens)