Update app.py

app.py

@@ -61,7 +61,7 @@ def infer(prompts, negative_prompts, image):
     ).images[0]
     
     #output_images = pipe.numpy_to_pil(np.asarray(output.reshape((num_samples,) + output.shape[-3:])))
-    #return output_images
+    return output
 
 with gr.Blocks(theme='kfahn/AnimalPose') as demo:  
   gr.Markdown(