gradio

app.py

@@ -65,50 +65,47 @@ def generate(
 #Move the components in the example fields outside so they are available when gr.Examples is instantiated
 
 with gr.Blocks() as demo:
-    gr.Markdown("<h1 style='text-align: center'>Omni Zero</h1>")
+    gr.Markdown("<h1 style='text-align: center'>Omni Zero Couples</h1>")
     gr.Markdown("<h4 style='text-align: center'>A diffusion pipeline for zero-shot stylized portrait creation [<a href='https://github.com/okaris/omni-zero' target='_blank'>GitHub</a>], [<a href='https://styleof.com/s/remix-yourself' target='_blank'>StyleOf Remix Yourself</a>]</h4>")
     with gr.Row():
         with gr.Column():
             with gr.Row():
-                prompt = gr.Textbox(label="Prompt", value="A person")
+                prompt = gr.Textbox(label="Prompt", value="Cinematic still photo of a couple. emotional, harmonious, vignette, 4k epic detailed, shot on kodak, 35mm photo, sharp focus, high budget, cinemascope, moody, epic, gorgeous, film grain, grainy")
             with gr.Row():
-                negative_prompt = gr.Textbox(label="Negative Prompt", value="blurry, out of focus")
+                negative_prompt = gr.Textbox(label="Negative Prompt", value="anime, cartoon, graphic, (blur, blurry, bokeh), text, painting, crayon, graphite, abstract, glitch, deformed, mutated, ugly, disfigured")
             with gr.Row():
                 with gr.Column(min_width=140):
                     with gr.Row():
-                        composition_image = gr.Image(label="Composition")
+                        base_image = gr.Image(label="Base Image")
                     with gr.Row():
-                        composition_image_strength = gr.Slider(label="Strength",step=0.01, minimum=0.0, maximum=1.0, value=1.0)
+                        base_image_strength = gr.Slider(label="Strength",step=0.01, minimum=0.0, maximum=1.0, value=1.0)
             #with gr.Row():
                 with gr.Column(min_width=140):
                     with gr.Row():
                         style_image = gr.Image(label="Style Image")
                     with gr.Row():
                         style_image_strength = gr.Slider(label="Strength",step=0.01, minimum=0.0, maximum=1.0, value=1.0)
+            with gr.Row():
                 with gr.Column(min_width=140):
                     with gr.Row():
                         identity_image = gr.Image(label="Identity Image")
                     with gr.Row():
                         identity_image_strength = gr.Slider(label="Strenght",step=0.01, minimum=0.0, maximum=1.0, value=1.0)
-            with gr.Accordion("Advanced options", open=False):
-                with gr.Row():
-                    with gr.Column(min_width=140):
-                            with gr.Row():
-                                base_image = gr.Image(label="Base Image")
-                            with gr.Row():
-                                base_image_strength = gr.Slider(label="Strength",step=0.01, minimum=0.0, maximum=1.0, value=0.15, min_width=120)
-                #    with gr.Column(min_width=140):
-                #     with gr.Row():
-                #         depth_image = gr.Image(label="depth_image", value=None)
-                #     with gr.Row():
-                #         depth_image_strength = gr.Slider(label="depth_image_strength",step=0.01, minimum=0.0, maximum=1.0, value=0.5)
-                            
+                with gr.Column(min_width=140):
+                    with gr.Row():
+                        identity_image_2 = gr.Image(label="Identity Image 2")
+                    with gr.Row():
+                        identity_image_strength_2 = gr.Slider(label="Strenght",step=0.01, minimum=0.0, maximum=1.0, value=1.0)
+            with gr.Accordion("Advanced options", open=False):      
                 with gr.Row():
                     seed = gr.Slider(label="Seed",step=1, minimum=0, maximum=10000000, value=42)
                     number_of_images = gr.Slider(label="Number of Outputs",step=1, minimum=1, maximum=4, value=1)
                 with gr.Row():
                     guidance_scale = gr.Slider(label="Guidance Scale",step=0.1, minimum=0.0, maximum=14.0, value=3.0)
                     number_of_steps = gr.Slider(label="Number of Steps",step=1, minimum=1, maximum=50, value=10)
+                with gr.Row():
+                    mask_guidance_start = gr.Slider(label="Mask Guidance Start",step=0.01, minimum=0.0, maximum=1.0, value=0.0)
+                    mask_guidance_end = gr.Slider(label="Mask Guidance End",step=0.01, minimum=0.0, maximum=1.0, value=1.0)
             
         with gr.Column():
             with gr.Row():
@@ -119,26 +116,36 @@ with gr.Blocks() as demo:
         
                 submit.click(generate, inputs=[
                     prompt,
-                    composition_image,
+                    base_image,
                     style_image,
                     identity_image,
-                    base_image,
+                    identity_image_2,
                     seed,
                     negative_prompt,
                     guidance_scale,
                     number_of_images,
                     number_of_steps,
                     base_image_strength,
-                    composition_image_strength,
                     style_image_strength,
                     identity_image_strength,
+                    identity_image_strength_2,
+                    mask_guidance_start,
+                    mask_guidance_end,
                     ],
                     outputs=[out]
                 )
         # clear.click(lambda: None, None, chatbot, queue=False)
     gr.Examples(
-        examples=[["A person", "https://github.com/okaris/omni-zero/assets/1448702/2ca63443-c7f3-4ba6-95c1-2a341414865f", "https://github.com/okaris/omni-zero/assets/1448702/64dc150b-f683-41b1-be23-b6a52c771584", "https://github.com/okaris/omni-zero/assets/1448702/ba193a3a-f90e-4461-848a-560454531c58"]],
-        inputs=[prompt, composition_image, style_image, identity_image],
+        examples=[
+            [
+                "Cinematic still photo of a couple. emotional, harmonious, vignette, 4k epic detailed, shot on kodak, 35mm photo, sharp focus, high budget, cinemascope, moody, epic, gorgeous, film grain, grainy",
+                "https://cdn-prod.styleof.com/inferences/cm1ho5cjl14nh14jec6phg2h8/i6k59e7gpsr45ufc7l8kun0g-medium.jpeg",
+                "https://cdn-prod.styleof.com/inferences/cm1ho5cjl14nh14jec6phg2h8/i6k59e7gpsr45ufc7l8kun0g-medium.jpeg",
+                "https://cdn-prod.styleof.com/inferences/cm1hp4lea14oz14jeoghnex7g/dlgc5xwo0qzey7qaixy45i1o-medium.jpeg",
+                "https://cdn-prod.styleof.com/inferences/cm1ho69ha14np14jesnusqiep/mp3aaktzqz20ujco5i3bi5s1-medium.jpeg"
+            ]
+        ],
+        inputs=[prompt, base_image, style_image, identity_image, identity_image_2],
         outputs=[out],
         fn=generate,
         cache_examples="lazy",