Update app.py

app.py

@@ -60,6 +60,8 @@ def generate_image(prompt: str, negative_prompt: str = "", inference_steps: int
             guidance_scale=float(guidance_scale),
             jit=True,
         ).images
+    images = images.reshape((num_samples,) + images.shape[-3:])
+    images = pipeline.numpy_to_pil(images)
     return image[0]
 
 examples = [
@@ -292,7 +294,7 @@ with block as demo:
                     fn=generate_image,
                     inputs=[prompt_input, negative,inf_steps_input, seed_input, guidance_scale],
                     outputs=[gallery],
-                    cache_examples=True)
+                    cache_examples=False)
                 ex.dataset.headers = [""]
 
                 share_button = gr.Button("Share to community",elem_classes="share_button")