Update app.py

app.py

@@ -67,7 +67,8 @@ def sample_ddpm_inference(text_prompt):
     """
 
     mask_image_pil = None
-    guidance_scale = 3.0
+    guidance_scale = 1.0
+    image_display_rate = 2
 
     # Create noise scheduler
     scheduler = LinearNoiseScheduler(
@@ -172,7 +173,7 @@ def sample_ddpm_inference(text_prompt):
                 noise_pred = noise_pred_cond
             xt, _ = scheduler.sample_prev_timestep(xt, noise_pred, t)
 
-            if i % 10 == 0 or i == 0:
+            if i % image_display_rate == 0 or i == 0:
                 # Decode current latent into image
                 generated = vae.decode(xt)
 
@@ -220,4 +221,4 @@ with gr.Blocks(css=css_str) as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch(share=True)
+    demo.launch(share=True)