Update app.py

app.py

@@ -43,17 +43,17 @@ flux_generator = FluxGenerator()
 @spaces.GPU
 @torch.inference_mode()
 def generate_image(
-        width,
-        height,
-        num_steps,
+        prompt,
+        id_image,
         start_step,
         guidance,
         seed,
-        prompt,
-        id_image=None,
+        true_cfg,
+        width=896,
+        height=1152,
+        num_steps=20,
         id_weight=1.0,
-        neg_prompt="",
-        true_cfg=1.0,
+        neg_prompt="bad quality, worst quality, text, signature, watermark, extra limbs",
         timestep_to_start_cfg=1,
         max_sequence_length=128,
 ):
@@ -154,18 +154,16 @@ _HEADER_ = '''
     <h1 style="font-size: 2.5rem; font-weight: 700; margin-bottom: 1rem; display: contents;">PuLID for FLUX</h1>
     <p style="font-size: 1rem; margin-bottom: 1.5rem;">Paper: <a href='https://arxiv.org/abs/2404.16022' target='_blank'>PuLID: Pure and Lightning ID Customization via Contrastive Alignment</a> | Codes: <a href='https://github.com/ToTheBeginning/PuLID' target='_blank'>GitHub</a></p>
 </div>
-
 ❗️❗️❗️**Tips:**
 - `timestep to start inserting ID:` The smaller the value, the higher the fidelity, but the lower the editability; the higher the value, the lower the fidelity, but the higher the editability. **The recommended range for this value is between 0 and 4**. For photorealistic scenes, we recommend using 4; for stylized scenes, we recommend using 0-1. If you are not satisfied with the similarity, you can lower this value; conversely, if you are not satisfied with the editability, you can increase this value.
 - `true CFG scale:` In most scenarios, it is recommended to use a fake CFG, i.e., setting the true CFG scale to 1, and just adjusting the guidance scale. This is also more efficiency. However, in a few cases, utilizing a true CFG can yield better results. For more detaileds, please refer to the [doc](https://github.com/ToTheBeginning/PuLID/blob/main/docs/pulid_for_flux.md#useful-tips).
-- please refer to the <a href='https://github.com/ToTheBeginning/PuLID/blob/main/docs/pulid_for_flux.md' target='_blank'>github doc</a> for more details and info about the model, we provide the detail explanation about the above two parameters in the doc.
-- we provide some examples in the bottom, you can try these example prompts first
+- `Learn more about the model:` please refer to the <a href='https://github.com/ToTheBeginning/PuLID/blob/main/docs/pulid_for_flux.md' target='_blank'>github doc</a> for more details and info about the model, we provide the detail explanation about the above two parameters in the doc.
+- `Examples:` we provide some examples in the bottom, you can try these example prompts first
 '''  # noqa E501
 
 _CITE_ = r"""
 If PuLID is helpful, please help to ⭐ the <a href='https://github.com/ToTheBeginning/PuLID' target='_blank'> Github Repo</a>. Thanks!
 ---
-
 📧 **Contact**
 If you have any questions or feedbacks, feel free to open a discussion or contact <b>[email protected]</b>.
 """  # noqa E501
@@ -217,22 +215,22 @@ def create_demo(args, model_name: str, device: str = "cuda" if torch.cuda.is_ava
                     [
                         'portrait, side view',
                         'example_inputs/liuyifei.png',
-                        4, 4, 1205240166692517553, 1
+                        4, 4, 180825677246321775, 1
                     ],
                     [
                         'white-haired woman with vr technology atmosphere, revolutionary exceptional magnum with remarkable details',  # noqa E501
                         'example_inputs/liuyifei.png',
-                        4, 4, 6349424134217931066, 1
+                        4, 4, 16942328329935464989, 1
                     ],
                     [
                         'a young child is eating Icecream',
                         'example_inputs/liuyifei.png',
-                        4, 4, 10606046113565776207, 1
+                        4, 4, 4527590969012358757, 1
                     ],
                     [
                         'a man is holding a sign with text \"PuLID for FLUX\", winter, snowing, top of the mountain',
                         'example_inputs/pengwei.jpg',
-                        4, 4, 2410129802683836089, 1
+                        4, 4, 6273700647573240909, 1
                     ],
                     [
                         'portrait, candle light',
@@ -256,22 +254,24 @@ def create_demo(args, model_name: str, device: str = "cuda" if torch.cuda.is_ava
                     ],
                 ]
                 gr.Examples(examples=example_inps, inputs=[prompt, id_image, start_step, guidance, seed, true_cfg],
-                            label='fake CFG')
+                            label='fake CFG', cache_examples='lazy', outputs=[output_image, seed_output],
+                            fn=generate_image)
 
                 example_inps = [
                     [
                         'portrait, made of ice sculpture',
                         'example_inputs/lecun.jpg',
-                        1, 1, 3811899118709451814, 5
+                        1, 1, 7717391560531186077, 5
                     ],
                 ]
                 gr.Examples(examples=example_inps, inputs=[prompt, id_image, start_step, guidance, seed, true_cfg],
-                            label='true CFG')
+                            label='true CFG', cache_examples='lazy', outputs=[output_image, seed_output],
+                            fn=generate_image)
 
         generate_btn.click(
             fn=generate_image,
-            inputs=[width, height, num_steps, start_step, guidance, seed, prompt, id_image, id_weight, neg_prompt,
-                    true_cfg, timestep_to_start_cfg, max_sequence_length],
+            inputs=[prompt, id_image, start_step, guidance, seed, true_cfg, width, height, num_steps, id_weight,
+                    neg_prompt, timestep_to_start_cfg, max_sequence_length],
             outputs=[output_image, seed_output, intermediate_output],
         )
 
@@ -296,4 +296,4 @@ if __name__ == "__main__":
     huggingface_hub.login(os.getenv('HF_TOKEN'))
 
     demo = create_demo(args, args.name, args.device, args.offload)
-    demo.launch()
+    demo.launch()