franciszzj commited on
Commit
ee88584
β€’
1 Parent(s): dd14542

update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -101,10 +101,12 @@ if __name__ == "__main__":
101
  # leffa_predict(src_image_path, ref_image_path, control_type)
102
 
103
  title = "## Leffa: Learning Flow Fields in Attention for Controllable Person Image Generation"
 
104
  description = "Leffa is a unified framework for controllable person image generation that enables precise manipulation of both appearance (i.e., virtual try-on) and pose (i.e., pose transfer)."
105
 
106
- with gr.Blocks(theme=gr.themes.Default(primary_hue=gr.themes.colors.pink, secondary_hue=gr.themes.colors.red)) as demo:
107
  gr.Markdown(title)
 
108
  gr.Markdown(description)
109
 
110
  with gr.Tab("Control Appearance (Virtual Try-on)"):
@@ -219,4 +221,4 @@ if __name__ == "__main__":
219
  pose_transfer_gen_button.click(fn=leffa_predict_pt, inputs=[
220
  pt_src_image, pt_ref_image], outputs=[pt_gen_image])
221
 
222
- demo.queue().launch(share=True, show_error=True)
 
101
  # leffa_predict(src_image_path, ref_image_path, control_type)
102
 
103
  title = "## Leffa: Learning Flow Fields in Attention for Controllable Person Image Generation"
104
+ link = "[πŸ“š Paper](https://arxiv.org/abs/2412.08486) - [πŸ”₯ Demo](https://huggingface.co/spaces/franciszzj/Leffa) - [πŸ€— Model](https://huggingface.co/franciszzj/Leffa)"
105
  description = "Leffa is a unified framework for controllable person image generation that enables precise manipulation of both appearance (i.e., virtual try-on) and pose (i.e., pose transfer)."
106
 
107
+ with gr.Blocks(theme=gr.themes.Default(primary_hue=gr.themes.colors.pink, secondary_hue=gr.themes.colors.red)).queue() as demo:
108
  gr.Markdown(title)
109
+ gr.Markdown(link)
110
  gr.Markdown(description)
111
 
112
  with gr.Tab("Control Appearance (Virtual Try-on)"):
 
221
  pose_transfer_gen_button.click(fn=leffa_predict_pt, inputs=[
222
  pt_src_image, pt_ref_image], outputs=[pt_gen_image])
223
 
224
+ demo.launch(share=True, server_port=7860)