update

app.py

@@ -98,18 +98,18 @@ def generate_image(
     ref_task1,
     ref_task2,
     prompt,
-    width,
-    height,
-    ref_res,
-    num_steps,
-    guidance,
-    seed,
-    true_cfg,
-    cfg_start_step,
-    cfg_end_step,
-    neg_prompt,
-    neg_guidance,
-    first_step_guidance,
+    width=1024,
+    height=1024,
+    ref_res=512,
+    num_steps=12,
+    guidance=3.5,
+    seed='-1',
+    true_cfg=1,
+    cfg_start_step=0,
+    cfg_end_step=0,
+    neg_prompt='',
+    neg_guidance=3.5,
+    first_step_guidance=0,
 ):
     print(prompt)
     ref_conds = []
@@ -224,30 +224,38 @@ def create_demo():
             example_inps = [
                 [
                     'example_inputs/woman1.png',
+                    None,
+                    'ip',
                     'ip',
                     'profile shot dark photo of a 25-year-old female with smoke escaping from her mouth, the backlit smoke gives the image an ephemeral quality, natural face, natural eyebrows, natural skin texture, award winning photo, highly detailed face, atmospheric lighting, film grain, monochrome',  # noqa E501
                     9180879731249039735,
                 ],
                 [
                     'example_inputs/man1.png',
+                    None,
+                    'ip',
                     'ip',
                     'a man sitting on the cloud, playing guitar',
                     1206523688721442817,
                 ],
                 [
                     'example_inputs/toy1.png',
+                    None,
+                    'ip',
                     'ip',
                     'a purple toy holding a sign saying "DreamO", on the mountain',
                     1563188099017016129,
                 ],
                 [
                     'example_inputs/perfume.png',
+                    None,
+                    'ip',
                     'ip',
                     'a perfume under spotlight',
                     116150031980664704,
                 ],
             ]
-            gr.Examples(examples=example_inps, inputs=[ref_image1, ref_task1, prompt, seed], label='IP task', cache_examples='lazy',
+            gr.Examples(examples=example_inps, inputs=[ref_image1, ref_image2, ref_task1, ref_task2, prompt, seed], label='IP task', cache_examples='lazy',
                         outputs=[output_image, debug_image, seed_output],
                         fn=generate_image,)
 
@@ -263,7 +271,7 @@ def create_demo():
             ]
             gr.Examples(
                 examples=example_inps,
-                inputs=[ref_image1, ref_task1, prompt, seed],
+                inputs=[ref_image1, ref_image2, ref_task1, ref_task2, prompt, seed],
                 label='ID task (similar to PuLID, will only refer to the face)',
                 cache_examples='lazy',
                 outputs=[output_image, debug_image, seed_output],
@@ -273,18 +281,22 @@ def create_demo():
             example_inps = [
                 [
                     'example_inputs/mickey.png',
+                    None,
                     'style',
+                    'ip',
                     'generate a same style image. A rooster wearing overalls.',
                     6245580464677124951,
                 ],
                 [
                     'example_inputs/mountain.png',
+                    None,
                     'style',
+                    'ip',
                     'generate a same style image. A pavilion by the river, and the distant mountains are endless',
                     5248066378927500767,
                 ],
             ]
-            gr.Examples(examples=example_inps, inputs=[ref_image1, ref_task1, prompt, seed], label='Style task', cache_examples='lazy',
+            gr.Examples(examples=example_inps, inputs=[ref_image1, ref_image2, ref_task1, ref_task2, prompt, seed], label='Style task', cache_examples='lazy',
                         outputs=[output_image, debug_image, seed_output],
                         fn=generate_image,)