Update app.py

app.py

@@ -42,7 +42,7 @@ def infer(source_img, prompt):
     images_list = img_pipe([prompt] * 1, guidance_scale=7.5, num_inference_steps=25, init_image=source_image,  strength=0.75)
     images = []
 
-    for i, image in enumerate(images_list["sample"]):
+    for i, image in enumerate(images_list["images"]):
             images.append(image)    
     return images