Update app.py

app.py

@@ -55,9 +55,7 @@ def create_image(image_pil,
         guidance_scale=guidance_scale,
         num_inference_steps=num_inference_steps,
         generator=generator,
-    )
-    if not isinstance(image, Image.Image):
-        image = Image.fromarray(image)
+    ).images[0]
     
     return image