Update app.py

app.py

@@ -95,7 +95,10 @@ def generate(prompt,
                                      guidance_scale=guidance_scale, 
                                      scale=cur_scale,  seed=seed, num_inference_steps=steps, avg_diff=avg_diff) 
         images.append(image)
+        #hack for cache examples
         counter+=1
+        if counter == interm_steps:
+            break
     canvas = Image.new('RGB', (256*interm_steps, 256))
     for i, im in enumerate(images):
         canvas.paste(im.resize((256,256)), (256 * i, 0))