Update app.py

app.py

@@ -58,7 +58,7 @@ def run(*args):
 
 
 _MARKDOWN_ = """
-<h2><b>This is the demo for ToonMagev2 for Image to Image Stylization</b></h2>
+<h2><b>This is the demo for DyCA-HiFi for Image to Identity Customization</b></h2>
 
 **Tips**
 
@@ -72,7 +72,7 @@ _MARKDOWN_ = """
   
   Try out with different prompts using your image and do provide your feedback.
 
-Note: The image of the women is generated using AI, therefore it does not resemble any living person. 
+Note: Most of the images are generated using AI, therefore it does not resemble any living person. 
   
 **Demo by [Sunder Ali Khowaja](https://sander-ali.github.io) - [X](https://x.com/SunderAKhowaja) -[Github](https://github.com/sander-ali) -[Hugging Face](https://huggingface.co/SunderAli17)**
 """
@@ -93,7 +93,7 @@ theme = gr.themes.Soft(
 # """
 
 
-with gr.Blocks(title="ToonMagev2", theme = "Yntec/HaleyCH_Theme_Orange") as SAK:
+with gr.Blocks(title="DyCA-HiFi", theme = "Yntec/HaleyCH_Theme_Orange") as SAK:
     gr.Markdown(_MARKDOWN_)
     with gr.Row():
         with gr.Column():
@@ -107,27 +107,27 @@ with gr.Blocks(title="ToonMagev2", theme = "Yntec/HaleyCH_Theme_Orange") as SAK:
 
             neg_prompt = gr.Textbox(label="Negative Prompt", value=DEFAULT_NEGATIVE_PROMPT)
             with gr.Row():
-                id_scale = gr.Slider(label="ID scale", minimum=0, maximum=5, step=0.05, value=0.8, interactive=True)
+                # id_scale = gr.Slider(label="ID scale", minimum=0, maximum=5, step=0.05, value=0.8, interactive=True)
                 mode = gr.Dropdown(label="mode", choices=['fidelity', 'extremely style'], value='fidelity')
                 id_mix = gr.Checkbox(
                     label="ID Mix (if you want to mix two ID image, please turn this on, otherwise, turn this off)",
                     value=False,
                 )
-            steps = gr.Slider(label="Steps", value=4, minimum=1, maximum=8, step=1)
-            seed = gr.Slider(
-                label="Seed", value=42, minimum=np.iinfo(np.uint32).min, maximum=np.iinfo(np.uint32).max, step=1
-            )
-            n_samples = gr.Slider(label="Num samples", value=2, minimum=1, maximum=4, step=1)
-            scale = gr.Slider(
-                label="CFG, recommend value range [1, 1.5], 1 will be faster ",
-                value=1.2,
-                minimum=1,
-                maximum=1.5,
-                step=0.1,
-            )
-            with gr.Row():
-                H = gr.Slider(label="Height", value=1024, minimum=512, maximum=1280, step=64)
-                W = gr.Slider(label="Width", value=768, minimum=512, maximum=1280, step=64)
+            # steps = gr.Slider(label="Steps", value=4, minimum=1, maximum=8, step=1)
+            # seed = gr.Slider(
+            #     label="Seed", value=42, minimum=np.iinfo(np.uint32).min, maximum=np.iinfo(np.uint32).max, step=1
+            # )
+            # n_samples = gr.Slider(label="Num samples", value=2, minimum=1, maximum=4, step=1)
+            # scale = gr.Slider(
+            #     label="CFG, recommend value range [1, 1.5], 1 will be faster ",
+            #     value=1.2,
+            #     minimum=1,
+            #     maximum=1.5,
+            #     step=0.1,
+            # )
+            # with gr.Row():
+            #     H = gr.Slider(label="Height", value=1024, minimum=512, maximum=1280, step=64)
+            #     W = gr.Slider(label="Width", value=768, minimum=512, maximum=1280, step=64)
 
             gr.Markdown("## Examples")
             example_inps = [
@@ -139,39 +139,39 @@ with gr.Blocks(title="ToonMagev2", theme = "Yntec/HaleyCH_Theme_Orange") as SAK:
             ]
             gr.Examples(examples=example_inps, inputs=[prompt, face_image, mode], label='realistic')
 
-            example_inps = [
-                [
-                    'portrait, impressionist painting, loose brushwork, vibrant color, light and shadow play',
-                    'sample_img/sample_img_test24.jpg',
-                    'fidelity',
-                ]
-            ]
+            # example_inps = [
+            #     [
+            #         'portrait, impressionist painting, loose brushwork, vibrant color, light and shadow play',
+            #         'sample_img/sample_img_test24.jpg',
+            #         'fidelity',
+            #     ]
+            # ]
             gr.Examples(examples=example_inps, inputs=[prompt, face_image, mode], label='painting style')
 
             example_inps = [
                 [
-                    'portrait, flat papercut style, silhouette, clean cuts, paper, sharp edges, minimalist,color block,man',
-                    'sample_img/sample_img_test1.jpg',
+                    'make the man look like gandalf from lord of the rings in a ghibli style',
+                    'sample_img/test_image_4.jpg',
                     'fidelity',
                 ]
             ]
             gr.Examples(examples=example_inps, inputs=[prompt, face_image, mode], label='papercut style')
 
-            example_inps = [
-                [
-                    'woman,cartoon,solo,Popmart Blind Box, Super Mario, 3d',
-                    'sample_img/sample_img_test24.jpg',
-                    'fidelity',
-                ]
-            ]
-            gr.Examples(examples=example_inps, inputs=[prompt, face_image, mode], label='3d style')
-            example_inps = [
-                [
-                    'portrait, pixar',
-                    'sample_img/sample_img_test1.jpg',
-                    'fidelity',
-                ]
-            ]
+            # example_inps = [
+            #     [
+            #         'woman,cartoon,solo,Popmart Blind Box, Super Mario, 3d',
+            #         'sample_img/sample_img_test24.jpg',
+            #         'fidelity',
+            #     ]
+            # ]
+            # gr.Examples(examples=example_inps, inputs=[prompt, face_image, mode], label='3d style')
+            # example_inps = [
+            #     [
+            #         'portrait, pixar',
+            #         'sample_img/sample_img_test1.jpg',
+            #         'fidelity',
+            #     ]
+            # ]
             gr.Examples(examples=example_inps, inputs=[prompt, face_image, mode], label='portrait style')
             example_inps = [
                 [
@@ -184,8 +184,8 @@ with gr.Blocks(title="ToonMagev2", theme = "Yntec/HaleyCH_Theme_Orange") as SAK:
 
             example_inps = [
                 [
-                    'portrait, the legend of zelda, anime',
-                    'sample_img/image1.png',
+                    'Make the character casually sip a cup of coffee standing in the moonlight in Ghibli style',
+                    'sample_img/test_image_1.jpg',
                     'extremely style',
                 ]
             ]
@@ -211,15 +211,15 @@ with gr.Blocks(title="ToonMagev2", theme = "Yntec/HaleyCH_Theme_Orange") as SAK:
         supp_image1,
         prompt,
         neg_prompt,
-        id_scale,
+        id_scale = 0.8,
         mode,
         id_mix,
-        steps,
-        seed,
-        n_samples,
-        scale,
-        H,
-        W
+        steps = 4,
+        seed = 42,
+        n_samples = 2,
+        scale = 1.2,
+        H = 1024,
+        W = 768
     ]
     submit.click(fn=run, inputs=inps, outputs=[output, intermediate_output])