Update app.py

app.py

@@ -182,7 +182,7 @@ def infer(ref_style_file, style_description, caption, progress):
             lam_style=1, lam_txt_alignment=1.0,
             use_ddim_sampler=True,
         )
-        for (sampled_c, _, _) in progress.tqdm(sampling_c, total=extras.sampling_configs['timesteps']):
+        for (sampled_c, _, _) in progress.tqdm(tqdm(sampling_c, total=extras.sampling_configs['timesteps'])):
         #for i, (sampled_c, _, _) in enumerate(sampling_c, 1):
         #    if i % 5 == 0:  # Update progress every 5 steps
         #        progress(0.4 + 0.3 * (i / extras.sampling_configs['timesteps']), f"Stage C reverse process: step {i}/{extras.sampling_configs['timesteps']}")
@@ -199,7 +199,7 @@ def infer(ref_style_file, style_description, caption, progress):
                 unconditions_b, device=device, **extras_b.sampling_configs,
             )
             for i, (sampled_b, _, _) in enumerate(sampling_b, 1):
-                if i % 5 == 0:  # Update progress every 5 steps
+                if i % 1 == 0:  # Update progress every 1 step
                     progress(0.7 + 0.2 * (i / extras_b.sampling_configs['timesteps']), f"Stage B reverse process: step {i}/{extras_b.sampling_configs['timesteps']}")
                 sampled_b = sampled_b
             sampled = models_b.stage_a.decode(sampled_b).float()