Update injection_main_HF.py

injection_main_HF.py

@@ -630,19 +630,43 @@ if __name__ == "__main__":
             label="Seed",
             info="Random seed for the model",
         )
-        app = gr.Interface(
-            fn=style_transfer_app,
-            inputs=[
-                text_input,
-                image_input,
-                cfg_slider,
-                content_slider,
-                style_slider,
-                seed_slider,
-            ],
-            outputs=["image"],
-            title="Artist Interactive Demo",
-            examples=examples,
-            cache_examples=False
-        )
+        image_output= gr.Image(format="png")
+        
+        with gr.Blocks() as app:
+
+            with gr.Column():
+                gr.Markdown("Artist Interactive Demo")
+                with gr.row():
+                    with gr.Column():
+                        text_input.render()
+                        image_input.render()
+                        with gr.Accordion("Advanced settings"):
+                            with gr.Column():
+                                cfg_slider.render()
+                                content_slider.render()
+                                style_slider.render()
+                                seed_slider.render()
+                        submit_btn = gr.Button("Submit")
+                    with gr.Column():
+                        image_output.render()
+                        gr.Examples(
+                            examples = examples,
+                            inputs = [text_input, image_input],
+                            cache_examples=False
+                        )
+            submit_btn.click(
+                fn=style_transfer_app,
+                inputs=[
+                    text_input,
+                    image_input,
+                    cfg_slider,
+                    content_slider,
+                    style_slider,
+                    seed_slider,
+                ],
+                outputs=[image_output],
+                show_api=False
+            )
+            
+        
         app.launch(show_api=False, show_error=True)