Update app.py

app.py

@@ -46,6 +46,21 @@ guidance_scale = 3.0
 # A value of 1.0 is sharper, but sometimes results in grainy artifacts.
 upsample_temp = 0.997
 
+samples = diffusion.p_sample_loop(
+    model_fn,
+    (full_batch_size, 3, options["image_size"], options["image_size"]),
+    device=device,
+    clip_denoised=True,
+    progress=True,
+    model_kwargs=model_kwargs,
+    cond_fn=None,
+)[:batch_size]
+model.del_cache()
+
+# Show the output
+show_images(samples)
+
+
 import gradio as gr
 def generate_upsampled_image_from_text(prompt):
     # Set the prompt text