Update app.py

app.py

@@ -43,35 +43,25 @@ def run(image, src_style, src_prompt, prompts, shared_score_shift, shared_score_
     images_pil = [Image.fromarray((img * 255).astype(np.uint8)) for img in images_a]
     return images_pil
 
+
 with gr.Blocks() as demo:
     gr.Markdown('''# Welcome to Tonic's Stable Style Align 
     Here you can generate images with a style from a reference image using [transfer style from sdxl](https://huggingface.co/docs/diffusers/main/en/using-diffusers/sdxl). Add a reference picture, describe the style and add prompts to generate images in that style. It's the most interesting with your own art!''')
     
-    image_input = gr.Image(label="Reference image", type="pil")
-    style_input = gr.Textbox(label="Describe the reference style")
-    image_desc_input = gr.Textbox(label="Describe the reference image")
-    prompts_input = gr.Textbox(label="Prompts to generate images (separate with new lines)", lines=5)
-    shared_score_shift_input = gr.Slider(value=1.1, label="shared_score_shift", minimum=1.0, maximum=2.0, step=0.05)
-    shared_score_scale_input = gr.Slider(value=1.0, label="shared_score_scale", minimum=0.0, maximum=1.0, step=0.05)
-    guidance_scale_input = gr.Slider(value=10.0, label="guidance_scale", minimum=5.0, maximum=20.0, step=1)
-    num_inference_steps_input = gr.Slider(value=12, label="num_inference_steps", minimum=1, maximum=12, step=1)
-    large_input = gr.Checkbox(False, label="Large (1024x1024)")
-    seed_input = gr.Slider(value=0, label="seed (0 for random)", minimum=0, maximum=1000000, step=42)
-
     with gr.Row():
-        image_input.display()
+        image_input = gr.Image(label="Reference image", type="pil")
     with gr.Row():
-        style_input.display()
-        image_desc_input.display()
-        prompts_input.display()
+        style_input = gr.Textbox(label="Describe the reference style")
+        image_desc_input = gr.Textbox(label="Describe the reference image")
+        prompts_input = gr.Textbox(label="Prompts to generate images (separate with new lines)", lines=5)
     with gr.Accordion(label="Advanced Settings"):
         with gr.Row():
-            shared_score_shift_input.display()
-            shared_score_scale_input.display()
-            guidance_scale_input.display()
-            num_inference_steps_input.display()
-            large_input.display()
-            seed_input.display()
+            shared_score_shift_input = gr.Slider(value=1.1, label="shared_score_shift", minimum=1.0, maximum=2.0, step=0.05)
+            shared_score_scale_input = gr.Slider(value=1.0, label="shared_score_scale", minimum=0.0, maximum=1.0, step=0.05)
+            guidance_scale_input = gr.Slider(value=10.0, label="guidance_scale", minimum=5.0, maximum=20.0, step=1)
+            num_inference_steps_input = gr.Slider(value=12, label="num_inference_steps", minimum=1, maximum=12, step=1)
+            large_input = gr.Checkbox(False, label="Large (1024x1024)")
+            seed_input = gr.Slider(value=0, label="seed (0 for random)", minimum=0, maximum=1000000, step=42)
     with gr.Row():
         run_button = gr.Button("Generate Images")
     with gr.Row():