22

app.py

@@ -86,6 +86,7 @@ def prepare(prompt, is_polish_prompt):
 @spaces.GPU
 def inference(
     prompt,
+    negative_prompt,
     input_image,
     image_scale=1.0,
     control_mode='Canny',
@@ -134,7 +135,7 @@ def inference(
 
     image = pipe(
         prompt=prompt,
-        negative_prompt = "blurry ugly bad",
+        negative_prompt = negative_prompt,
         height=height,
         width=width,
         generator=generator,
@@ -178,7 +179,7 @@ with gr.Blocks(css=css) as demo:
                 
                 prompt = gr.Textbox(
                     label="Prompt",
-                    # show_label=False,
+                    show_label=False,
                     lines=2,
                     placeholder="Enter your prompt",
                     # container=False,
@@ -190,6 +191,54 @@ with gr.Blocks(css=css) as demo:
                     label="Control Mode"
                 )
                 run_button = gr.Button("Generate", variant="primary")
+                with gr.Accordion("Advanced Settings", open=False):
+                    seed = gr.Slider(
+                        label="Seed",
+                        minimum=0,
+                        maximum=MAX_SEED,
+                        step=1,
+                        value=42,
+                    )
+
+                    randomize_seed = gr.Checkbox(label="Randomize seed", value=False)
+                    negative_prompt = gr.Textbox(
+                        label="Negative prompt",
+                        lines=2,
+                        placeholder="Enter your prompt",
+                        value="blurry ugly bad"
+                    )
+                    with gr.Row():
+                        image_scale = gr.Slider(
+                            label="Image scale",
+                            minimum=0.5,
+                            maximum=2.0,
+                            step=0.1,
+                            value=1.0,
+                        )
+                        control_context_scale = gr.Slider(
+                            label="Control context scale",
+                            minimum=0.0,
+                            maximum=1.0,
+                            step=0.01,
+                            value=0.75,
+                        )
+
+                    with gr.Row():
+                        guidance_scale = gr.Slider(
+                            label="Guidance scale",
+                            minimum=0.0,
+                            maximum=10.0,
+                            step=0.1,
+                            value=1.0,
+                        )
+
+                        num_inference_steps = gr.Slider(
+                            label="Number of inference steps",
+                            minimum=1,
+                            maximum=50,
+                            step=1,
+                            value=15,
+                        )
             with gr.Column():
                 output_image = gr.Image(label="Generated image", show_label=False)
                 polished_prompt = gr.Textbox(label="Polished prompt", interactive=False)
@@ -198,49 +247,7 @@ with gr.Blocks(css=css) as demo:
                     control_image = gr.Image(label="Control image", show_label=False)
                     
 
-        with gr.Accordion("Advanced Settings", open=False):
-                seed = gr.Slider(
-                    label="Seed",
-                    minimum=0,
-                    maximum=MAX_SEED,
-                    step=1,
-                    value=42,
-                )
-
-                randomize_seed = gr.Checkbox(label="Randomize seed", value=False)
-
-                with gr.Row():
-                    image_scale = gr.Slider(
-                        label="Image scale",
-                        minimum=0.5,
-                        maximum=2.0,
-                        step=0.1,
-                        value=1.0,
-                    )
-                    control_context_scale = gr.Slider(
-                        label="Control context scale",
-                        minimum=0.0,
-                        maximum=1.0,
-                        step=0.01,
-                        value=0.75,
-                    )
-
-                with gr.Row():
-                    guidance_scale = gr.Slider(
-                        label="Guidance scale",
-                        minimum=0.0,
-                        maximum=10.0,
-                        step=0.1,
-                        value=1.0,
-                    )
-
-                    num_inference_steps = gr.Slider(
-                        label="Number of inference steps",
-                        minimum=1,
-                        maximum=50,
-                        step=1,
-                        value=15,
-                    )
+        
         gr.Examples(examples=examples, inputs=[input_image, prompt, control_mode])
         gr.HTML(read_file("static/footer.html"))
 
@@ -253,6 +260,7 @@ with gr.Blocks(css=css) as demo:
         fn=inference,
         inputs=[
             polished_prompt,
+            negative_prompt,
             input_image,
             image_scale,
             control_mode,