Update app.py

app.py

@@ -10,6 +10,7 @@ import re
 import gradio as gr
 import numpy as np
 import tempfile
+import tqdm
 from einops import rearrange
 from pydub import AudioSegment, silence
 from model import UNetT, DiT
@@ -259,8 +260,10 @@ def infer_batch(ref_audio, ref_text, gen_text_batches, exp_name, remove_silence,
 
 # Define weights for characters
     punctuation_weights = {",": 0, ".": 0, " ": 0}  # Add more punctuation as needed
+    progress = tqdm(gen_text_batches)
+    for i, gen_text in enumerate(progress):
 
-    for i, gen_text in enumerate(progress.tqdm(gen_text_batches)):
+   # for i, gen_text in enumerate(progress.tqdm(gen_text_batches)):
         # Prepare the text
         ipa_text_ref = text_to_ipa(ref_text, language=language)
         ipa_text_gen = text_to_ipa(gen_text, language=language)