bug fix

app.py

@@ -57,13 +57,12 @@ def inference(
 ):
     prompt = replace_concept_tokens(prompt)
     generator = torch.Generator(device=device).manual_seed(seed)
-    out = pipeline(
+    img_list = pipeline(
         prompt=[prompt] * 2,
         num_inference_steps=num_inference_steps,
         guidance_scale=guidance_scale,
         generator=generator,
-    )
-    img_list = [item['sample'] for item in out]
+    ).images
     return img_list
 
 DEFAULT_PROMPT = (