Fix the error of 'sample' key not found (#5)

- Fix the error of 'sample' key not found (137c815911a23a838611a269cae5870a033e1fa8)


Co-authored-by: Naoto Usuyama <[email protected]>

app.py

@@ -14,7 +14,7 @@ def predict(prompt, steps=100, seed=42, guidance_scale=6.0):
     #torch.cuda.empty_cache()
     print(subprocess.check_output(["nvidia-smi"], stderr=subprocess.STDOUT).decode("utf8"))
     generator = torch.manual_seed(seed)
-    images = ldm_pipeline([prompt], generator=generator, num_inference_steps=steps, eta=0.3, guidance_scale=guidance_scale)["sample"]
+    images = ldm_pipeline([prompt], generator=generator, num_inference_steps=steps, eta=0.3, guidance_scale=guidance_scale)["images"]
     print(subprocess.check_output(["nvidia-smi"], stderr=subprocess.STDOUT).decode("utf8"))
     return images[0]