Update app.py

app.py

@@ -427,7 +427,8 @@ def generate_image(
 
 # Description
 title = r"""
-<h1 align="center">InstantID: Zero-shot Identity-Preserving Generation in Seconds</h1>
+<h1 align="center">InstantID FaceID 6M: Zero-shot Identity-Preserving Generation in Seconds</h1>
+Demo for the `Super-shuhe/InstantID-FaceID-6M` trained on the [FaceID 6M](https://huggingface.co/datasets/Super-shuhe/FaceID-6M) open dataset of 6M faces
 """
 
 article = r"""
@@ -506,38 +507,6 @@ with gr.Blocks(css=css) as demo:
                     step=0.05,
                     value=0.80,
                 )
-            with gr.Accordion("Controlnet", open=False):
-                # optional: upload a reference pose image
-                pose_file = gr.Image(
-                    label="Upload a reference pose image (Optional)",
-                    type="filepath",
-                )
-                controlnet_selection = gr.CheckboxGroup(
-                    ["canny", "depth"], label="Controlnet", value=[],
-                    info="Use pose for skeleton inference, canny for edge detection, and depth for depth map estimation. You can try all three to control the generation process"
-                )
-                # pose_strength = gr.Slider(
-                #     label="Pose strength",
-                #     minimum=0,
-                #     maximum=1.5,
-                #     step=0.05,
-                #     value=0.40,
-                # )
-                canny_strength = gr.Slider(
-                    label="Canny strength",
-                    minimum=0,
-                    maximum=1.5,
-                    step=0.05,
-                    value=0,
-                )
-                depth_strength = gr.Slider(
-                    label="Depth strength",
-                    minimum=0,
-                    maximum=1.5,
-                    step=0.05,
-                    value=0,
-                )
-            with gr.Accordion(open=False, label="Advanced Options"):
                 negative_prompt = gr.Textbox(
                     label="Negative Prompt",
                     placeholder="low quality",
@@ -579,6 +548,38 @@ with gr.Blocks(css=css) as demo:
                 )
                 randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
                 enhance_face_region = gr.Checkbox(label="Enhance non-face region", value=True)
+                
+            with gr.Accordion("Controlnet", open=False):
+                # optional: upload a reference pose image
+                pose_file = gr.Image(
+                    label="Upload a reference pose image (Optional)",
+                    type="filepath",
+                )
+                controlnet_selection = gr.CheckboxGroup(
+                    ["canny", "depth"], label="Controlnet", value=[],
+                    info="Use pose for skeleton inference, canny for edge detection, and depth for depth map estimation. You can try all three to control the generation process"
+                )
+                # pose_strength = gr.Slider(
+                #     label="Pose strength",
+                #     minimum=0,
+                #     maximum=1.5,
+                #     step=0.05,
+                #     value=0.40,
+                # )
+                canny_strength = gr.Slider(
+                    label="Canny strength",
+                    minimum=0,
+                    maximum=1.5,
+                    step=0.05,
+                    value=0,
+                )
+                depth_strength = gr.Slider(
+                    label="Depth strength",
+                    minimum=0,
+                    maximum=1.5,
+                    step=0.05,
+                    value=0,
+                )
 
         with gr.Column(scale=1):
             gallery = gr.Image(label="Generated Images")