Update app.py

app.py

@@ -56,8 +56,12 @@ def create_image(image_pil,
         num_inference_steps=num_inference_steps,
         generator=generator,
     )
+    if not isinstance(image, Image.Image):
+        image = Image.fromarray(image)
+    
     return image
     
+    
 
 # Description
 title = r"""