Update app.py

app.py

@@ -124,6 +124,7 @@ def apply_style(style_name: str, positive: str, negative: str = "") -> tuple[str
     p, n = styles.get(style_name, styles[DEFAULT_STYLE_NAME])
     return p.replace("{prompt}", positive), n + negative
 
+
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 eulera_scheduler = EulerAncestralDiscreteScheduler.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", subfolder="scheduler")
@@ -237,7 +238,7 @@ def run(
         pipe = pipe_cyber
     else:  # "Stallion Dreams Pony Realistic v1"
         pipe = pipe_stallion
-
+    
     if use_canny:
         out = pipe(
             prompt=prompt,
@@ -278,6 +279,7 @@ with gr.Blocks(css="style.css", js=js_func) as demo:
             with gr.Group():
                 image = gr.ImageEditor(type="pil", label="Sketch your image or upload one", width=512, height=512)
                 prompt = gr.Textbox(label="Prompt")
+                style = gr.Dropdown(label="Style", choices=STYLE_NAMES, value=DEFAULT_STYLE_NAME)
                 model_choice = gr.Dropdown(
                     ["Pony Realism v21", "Cyber Realistic Pony v61", "Stallion Dreams Pony Realistic v1"],
                     label="Model Choice",
@@ -330,8 +332,8 @@ with gr.Blocks(css="style.css", js=js_func) as demo:
         image,
         prompt,
         negative_prompt,
-        model_choice,
-        style,  # Replace 'style_name' with 'style'
+        model_choice,  # Add this new input
+        style,
         num_steps,
         guidance_scale,
         controlnet_conditioning_scale,
@@ -339,7 +341,6 @@ with gr.Blocks(css="style.css", js=js_func) as demo:
         use_hed,
         use_canny
     ]
-
     outputs = [image_slider]
     run_button.click(
         fn=randomize_seed_fn,