定义API接口

app.py

@@ -319,113 +319,113 @@ def generate_image(
 # .gradio-container {width: 85% !important}
 # """
 with gr.Blocks(css=css) as demo:
-#     # description
-#     gr.Markdown(title)
-#     gr.Markdown(description)
-
-#     with gr.Row():
-#         with gr.Column():
-#             # upload face image
-#             face_file = gr.Image(label="Upload a photo of your face", type="filepath")
-
-#             # optional: upload a reference pose image
-#             pose_file = gr.Image(label="Upload a reference pose image (optional)", type="filepath")
-
-#             # prompt
-#             prompt = gr.Textbox(
-#                 label="Prompt",
-#                 info="Give simple prompt is enough to achieve good face fedility",
-#                 placeholder="A photo of a person",
-#                 value="",
-#             )
-
-#             submit = gr.Button("Submit", variant="primary")
-
-#             style = gr.Dropdown(label="Style template", choices=STYLE_NAMES, value=DEFAULT_STYLE_NAME)
-
-#             # strength
-#             identitynet_strength_ratio = gr.Slider(
-#                 label="IdentityNet strength (for fedility)",
-#                 minimum=0,
-#                 maximum=1.5,
-#                 step=0.05,
-#                 value=0.80,
-#             )
-#             adapter_strength_ratio = gr.Slider(
-#                 label="Image adapter strength (for detail)",
-#                 minimum=0,
-#                 maximum=1.5,
-#                 step=0.05,
-#                 value=0.80,
-#             )
-
-#             with gr.Accordion(open=False, label="Advanced Options"):
-#                 negative_prompt = gr.Textbox(
-#                     label="Negative Prompt",
-#                     placeholder="low quality",
-#                     value="(lowres, low quality, worst quality:1.2), (text:1.2), watermark, (frame:1.2), deformed, ugly, deformed eyes, blur, out of focus, blurry, deformed cat, deformed, nudity,naked, bikini, skimpy, scanty, bare skin, lingerie, swimsuit, exposed, see-through, photo, anthropomorphic cat, monochrome, pet collar, gun, weapon, blue, 3d, drones, drone, buildings in background, green",
-#                 )
-#                 num_steps = gr.Slider(
-#                     label="Number of sample steps",
-#                     minimum=20,
-#                     maximum=100,
-#                     step=1,
-#                     value=30,
-#                 )
-#                 guidance_scale = gr.Slider(
-#                     label="Guidance scale",
-#                     minimum=0.1,
-#                     maximum=10.0,
-#                     step=0.1,
-#                     value=5,
-#                 )
-#                 seed = gr.Slider(
-#                     label="Seed",
-#                     minimum=0,
-#                     maximum=MAX_SEED,
-#                     step=1,
-#                     value=42,
-#                 )
-#                 randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
-#                 enhance_face_region = gr.Checkbox(label="Enhance non-face region", value=True)
-
-#         with gr.Column():
-#             output_image = gr.Image(label="Generated Image")
-#             usage_tips = gr.Markdown(label="Usage tips of InstantID", value=tips, visible=False)
-
-#         submit.click(
-#             fn=remove_tips,
-#             outputs=usage_tips,
-#             queue=False,
-#             api_name=False,
-#         ).then(
-#             fn=randomize_seed_fn,
-#             inputs=[seed, randomize_seed],
-#             outputs=seed,
-#             queue=False,
-#             api_name=False,
-#         ).then(
-#             fn=check_input_image,
-#             inputs=face_file,
-#             queue=False,
-#             api_name=False,
-#         ).success(
-#             fn=generate_image,
-#             inputs=[
-#                 face_file,
-#                 pose_file,
-#                 prompt,
-#                 negative_prompt,
-#                 style,
-#                 enhance_face_region,
-#                 num_steps,
-#                 identitynet_strength_ratio,
-#                 adapter_strength_ratio,
-#                 guidance_scale,
-#                 seed,
-#             ],
-#             outputs=[output_image, usage_tips],
-#         )
+    # description
+    # gr.Markdown(title)
+    # gr.Markdown(description)
+
+    with gr.Row():
+        with gr.Column():
+            # upload face image
+            face_file = gr.Image(label="Upload a photo of your face", type="filepath")
+
+            # optional: upload a reference pose image
+            pose_file = gr.Image(label="Upload a reference pose image (optional)", type="filepath")
+
+            # prompt
+            prompt = gr.Textbox(
+                label="Prompt",
+                info="Give simple prompt is enough to achieve good face fedility",
+                placeholder="A photo of a person",
+                value="",
+            )
+
+            submit = gr.Button("Submit", variant="primary")
+
+            style = gr.Dropdown(label="Style template", choices=STYLE_NAMES, value=DEFAULT_STYLE_NAME)
+
+            # strength
+            identitynet_strength_ratio = gr.Slider(
+                label="IdentityNet strength (for fedility)",
+                minimum=0,
+                maximum=1.5,
+                step=0.05,
+                value=0.80,
+            )
+            adapter_strength_ratio = gr.Slider(
+                label="Image adapter strength (for detail)",
+                minimum=0,
+                maximum=1.5,
+                step=0.05,
+                value=0.80,
+            )
+
+            with gr.Accordion(open=False, label="Advanced Options"):
+                negative_prompt = gr.Textbox(
+                    label="Negative Prompt",
+                    placeholder="low quality",
+                    value="(lowres, low quality, worst quality:1.2), (text:1.2), watermark, (frame:1.2), deformed, ugly, deformed eyes, blur, out of focus, blurry, deformed cat, deformed, nudity,naked, bikini, skimpy, scanty, bare skin, lingerie, swimsuit, exposed, see-through, photo, anthropomorphic cat, monochrome, pet collar, gun, weapon, blue, 3d, drones, drone, buildings in background, green",
+                )
+                num_steps = gr.Slider(
+                    label="Number of sample steps",
+                    minimum=20,
+                    maximum=100,
+                    step=1,
+                    value=30,
+                )
+                guidance_scale = gr.Slider(
+                    label="Guidance scale",
+                    minimum=0.1,
+                    maximum=10.0,
+                    step=0.1,
+                    value=5,
+                )
+                seed = gr.Slider(
+                    label="Seed",
+                    minimum=0,
+                    maximum=MAX_SEED,
+                    step=1,
+                    value=42,
+                )
+                randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
+                enhance_face_region = gr.Checkbox(label="Enhance non-face region", value=True)
+
+        with gr.Column():
+            output_image = gr.Image(label="Generated Image")
+            usage_tips = gr.Markdown(label="Usage tips of InstantID", value=tips, visible=False)
+
+        submit.click(
+            fn=remove_tips,
+            outputs=usage_tips,
+            queue=False,
+            api_name=False,
+        ).then(
+            fn=randomize_seed_fn,
+            inputs=[seed, randomize_seed],
+            outputs=seed,
+            queue=False,
+            api_name=False,
+        ).then(
+            fn=check_input_image,
+            inputs=face_file,
+            queue=False,
+            api_name=False,
+        ).success(
+            fn=generate_image,
+            inputs=[
+                face_file,
+                pose_file,
+                prompt,
+                negative_prompt,
+                style,
+                enhance_face_region,
+                num_steps,
+                identitynet_strength_ratio,
+                adapter_strength_ratio,
+                guidance_scale,
+                seed,
+            ],
+            outputs=[output_image, usage_tips],
+        )
 
     # gr.Examples(
     #     examples=get_example(),