multimodalart HF staff commited on
Commit
872fe49
β€’
1 Parent(s): a550ff1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -44,8 +44,9 @@ def generate_image(image, prompt, negative_prompt):
44
 
45
  image = ip_model.generate(
46
  prompt=prompt, negative_prompt=negative_prompt, faceid_embeds=faceid_embeds, width=512, height=512, num_inference_steps=30
47
- ).images[0]
 
48
  return image
49
 
50
- 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")])
51
  demo.launch()
 
44
 
45
  image = ip_model.generate(
46
  prompt=prompt, negative_prompt=negative_prompt, faceid_embeds=faceid_embeds, width=512, height=512, num_inference_steps=30
47
+ )
48
+ print(image)
49
  return image
50
 
51
+ demo = gr.Interface(fn=generate_image, inputs=[gr.Image(label="Your face"), gr.Textbox(label="Prompt"), gr.Textbox(label="Negative Prompt")], outputs=[gr.Gallery(label="Generated Image")])
52
  demo.launch()