Update app.py

app.py

@@ -42,10 +42,10 @@ def generate_image(image, prompt, negative_prompt):
     faces = app.get(image)
     faceid_embeds = torch.from_numpy(faces[0].normed_embedding).unsqueeze(0)
     
-    images = ip_model.generate(
+    image = ip_model.generate(
         prompt=prompt, negative_prompt=negative_prompt, faceid_embeds=faceid_embeds, width=512, height=512, num_inference_steps=30
-    )
-    return images.image[0]
+    ).images[0]
+    return image
 
 demo = gr.Interface(fn=generate_image, inputs=[gr.Image(label="Your face"), gr.Textbox(label="Prompt"), gr.Textbox(label="Negative Prompt")], outputs=[gr.Image(label="Generated Image")])
 demo.launch()