Update handler.py

handler.py

@@ -47,7 +47,7 @@ class EndpointHandler():
                 latents_dist = vae.encode(latents).latent_dist.sample() * vae.config.scaling_factor
             
             self.smooth_pipe.enable_xformers_memory_efficient_attention()
-            out = self.smooth_pipe(prompt, image=latents_dist).images[0]
+            out = self.smooth_pipe(prompt, image=latents_dist).images
 
             return out