Upload 40 files

.gitattributes

@@ -52,3 +52,5 @@ assets/examples/3-stylegan2-ffhq-0381.png filter=lfs diff=lfs merge=lfs -text
 assets/examples/Trump-4.jpg filter=lfs diff=lfs merge=lfs -text
 assets/examples/1-output-4.png filter=lfs diff=lfs merge=lfs -text
 assets/examples/1-output-5.png filter=lfs diff=lfs merge=lfs -text
+assets/examples/1-output-7.png filter=lfs diff=lfs merge=lfs -text
+assets/examples/2-output-5.png filter=lfs diff=lfs merge=lfs -text

app.py

@@ -86,6 +86,28 @@ def text_to_single_id_generation_block():
             examples=examples,
             inputs=[prompt, pil_faceid, pil_style, example_output],
         )
+    with gr.Row():
+        examples = [
+            [
+                "A woman, vector art, flat illustration, pastel color",
+                "assets/examples/1-alibaba-sucai-1.jpg",
+                0.5,
+                0.0,
+                "assets/examples/1-output-6.png",
+            ],
+            [
+                "A woman, comic book style, pastel color",
+                "assets/examples/1-alibaba-sucai-2.jpg",
+                0.5,
+                0.0,
+                "assets/examples/1-output-7.png",
+            ],
+        ]
+        gr.Examples(
+            label="Advanced Examples",
+            examples=examples,
+            inputs=[prompt, pil_faceid, faceid_scale, face_structure_scale, example_output],
+        )
     ips = [
         pil_faceid, pil_supp_faceids,
         pil_mix_faceid_1, mix_scale_1,
@@ -161,6 +183,13 @@ def text_to_multi_id_generation_block():
                                             format="png")
     with gr.Row():
         examples = [
+            [
+                "Two beautiful women are drinking coffee.",
+                "assets/examples/1-alibaba-sucai-2.jpg",
+                "assets/examples/2-alibaba-sucai-1.jpg",
+                "assets/No-Image-Placeholder.png",
+                "assets/examples/2-output-5.png",
+            ],
             [
                 "The two women are captured in laughter and joy, their faces brimming with sincere happiness against the backdrop of a peaceful beach at sunset. The painting, depicted in soft style, captures the warmth and tranquility of the moment.",
                 "assets/examples/2-stylegan2-ffhq-0100.png",
@@ -177,7 +206,7 @@ def text_to_multi_id_generation_block():
             ],
         ]
         gr.Examples(
-            label="Examples",
+            label="Advanced Examples",
             examples=examples,
             inputs=[prompt, pil_faceid_1st, pil_faceid_2nd, pil_style, example_output],
         )
@@ -324,7 +353,7 @@ if __name__ == "__main__":
 
     title_description = r"""
         This is the <b>official 🤗 Gradio demo</b> for <a href='https://arxiv.org/pdf/2408.05939' target='_blank'><b>UniPortrait: A Unified Framework for Identity-Preserving Single- and Multi-Human Image Personalization</b></a>.<br>
-        The demo provides three capabilities: text-to-single-ID personalization, text-to-multi-ID personalization, and image-to-single-ID personalization. All of these are based on the Stable Diffusion v1-5 model. Feel free to give them a try! 😊
+        The demo provides three capabilities: text-to-single-ID personalization, text-to-multi-ID personalization, and image-to-single-ID personalization. All of these are based on the <b>Stable Diffusion v1-5</b> model. Feel free to give them a try! 😊
         """
 
     text_to_single_id_description = r"""🚀🚀🚀Quick start:<br>
@@ -363,10 +392,16 @@ if __name__ == "__main__":
         <br>
         If our work is helpful for your research or applications, please cite us via:
         ```bibtex
+        @article{he2024uniportrait,
+          title={UniPortrait: A Unified Framework for Identity-Preserving Single-and Multi-Human Image Personalization},
+          author={He, Junjie and Geng, Yifeng and Bo, Liefeng},
+          journal={arXiv preprint arXiv:2408.05939},
+          year={2024}
+        }
         ```
         📧 **Contact**
         <br>
-        If you have any questions, please feel free to open an issue or directly reach us out at <b>he_junjie@zju.edu.cn</b>.
+        If you have any questions, please feel free to open an issue or directly reach us out at <b>hejunjie1103@gmail.com</b>.
         """
 
     block = gr.Blocks(title="UniPortrait").queue()
@@ -383,6 +418,8 @@ if __name__ == "__main__":
         with gr.TabItem("Image-to-Single-ID (Stylization)"):
             image_to_single_id_generation_block()
 
+        gr.Markdown(citation)
+
     block.launch(share=True, max_threads=10)
     # block.launch(server_name='0.0.0.0', share=False, server_port=9999, allowed_paths=["/"])
     # block.launch(server_name='127.0.0.1', share=False, server_port=9999, allowed_paths=["/"])