Fix output of pipeline

app.py

@@ -120,7 +120,7 @@ def stable_diffusion_compose(prompt, scale, steps, weights, seed):
     generator = th.Generator("cuda").manual_seed(int(seed))
     with autocast('cpu' if not th.cuda.is_available() else 'cuda'):
         image = pipe(prompt, guidance_scale=scale, num_inference_steps=steps,
-                     weights=weights, generator=generator)["sample"][0]
+                     weights=weights, generator=generator).images[0]
         image.save(f'{"_".join(prompt.split())}.png')
         return image