Update app.py

app.py

@@ -101,14 +101,16 @@ def pipe_inference(
         guidance_scale=guidance_scale,
         neg_prompt_ids=negative_prompt_ids,
         jit=True,
-    )
-    all_outputs = []
-    all_outputs.append(image)
-    if not is_canny:
-        all_outputs.append(resized_image)
+    ).images
+    # all_outputs = []
+    # all_outputs.append(image)
+    # if not is_canny:
+    #     all_outputs.append(resized_image)
         
-    for image in output.images:
-        all_outputs.append(image)
+    # for image in output.images:
+    #     all_outputs.append(image)
+
+    all_outputs = pipe.numpy_to_pil(np.asarray(output.reshape((num_samples,) + output.shape[-3:])))
     return all_outputs
 
 def resize_image(image, resolution):