franciszzj commited on
Commit
9b1ec91
β€’
1 Parent(s): 3b5e117

update note

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -103,6 +103,7 @@ if __name__ == "__main__":
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)
@@ -221,4 +222,6 @@ if __name__ == "__main__":
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)
 
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
+ note = "Note: The models used in the demo are trained solely on academic datasets. Virtual try-on uses VITON-HD, and pose transfer uses DeepFashion."
107
 
108
  with gr.Blocks(theme=gr.themes.Default(primary_hue=gr.themes.colors.pink, secondary_hue=gr.themes.colors.red)).queue() as demo:
109
  gr.Markdown(title)
 
222
  pose_transfer_gen_button.click(fn=leffa_predict_pt, inputs=[
223
  pt_src_image, pt_ref_image], outputs=[pt_gen_image])
224
 
225
+ gr.Markdown(note)
226
+
227
  demo.launch(share=True, server_port=7860)