Update app.py

app.py

@@ -139,7 +139,7 @@ E2TTS_ema_model2 = load_custom(
 
 
 
-def chunk_text(text, max_chars=100):
+def chunk_text(text, max_chars):
     """
     Splits the input text into chunks, ensuring:
     - Chunks are split by punctuation where possible.
@@ -233,7 +233,7 @@ def infer_batch(ref_audio, ref_text, gen_text_batches, exp_name, remove_silence,
         ref_text = ref_text + " "
 
 # Define weights for characters
-    punctuation_weights = {",": 0, ".": 0, " ": 0}  # Add more punctuation as needed
+    punctuation_weights = {",": 2, ".": 2, " ": 2}  # Add more punctuation as needed
 
     for i, gen_text in enumerate(progress.tqdm(gen_text_batches)):
         # Prepare the text
@@ -416,7 +416,7 @@ def infer(ref_audio_orig, ref_text, gen_text, exp_name, remove_silence, cross_fa
     audio, sr = torchaudio.load(ref_audio)
 
     # Use the new chunk_text function to split gen_text
-    max_chars = int(0.5 * (len(ref_text.encode('utf-8')) / (audio.shape[-1] / sr) * (18 - audio.shape[-1] / sr )))
+    max_chars = int(speed * 0.5 * (len(ref_text.encode('utf-8')) / (audio.shape[-1] / sr) * (18 - audio.shape[-1] / sr )))
     print(f"text: {max_chars} ")
     gen_text_batches = chunk_text(gen_text, max_chars=max_chars)
     print('ref_text', ref_text)