Update injection_main_HF.py

injection_main_HF.py

@@ -586,51 +586,10 @@ if __name__ == "__main__":
 
             examples.append([entry["target_prompt"], image, None, None, None])
 
-        text_input = gr.Textbox(
-            value="An impressionist painting",
-            label="Text Prompt",
-            info="Describe the style you want to apply to the image, do not include the description of the image content itself",
-            lines=2,
-            placeholder="Enter a text prompt",
-        )
-        image_input = gr.Image(
-            height="80%",
-            width="80%",
-            label="Content image (will be resized to 512x512)",
-            interactive=True,
-        )
-        cfg_slider = gr.Slider(
-            0,
-            15,
-            value=7.5,
-            label="Classifier Free Guidance (CFG) Scale",
-            info="higher values give more style, 7.5 should be good for most cases",
-        )
-        content_slider = gr.Slider(
-            0,
-            9,
-            value=4,
-            step=1,
-            label="Number of content control layer",
-            info="higher values make it more similar to original image. Default to control first 4 layers",
-        )
-        style_slider = gr.Slider(
-            0,
-            9,
-            value=9,
-            step=1,
-            label="Number of style control layer",
-            info="higher values make it more similar to target style. Default to control first 9 layers, usually not necessary to change.",
-        )
-        seed_slider = gr.Slider(
-            0,
-            100,
-            value=0,
-            step=1,
-            label="Seed",
-            info="Random seed for the model",
-        )
-        image_output= gr.Image(format="png")
+       
+        
+        
+        
         
         with gr.Blocks() as app:
 
@@ -638,17 +597,53 @@ if __name__ == "__main__":
                 gr.Markdown("Artist Interactive Demo")
                 with gr.Row():
                     with gr.Column():
-                        text_input.render()
-                        image_input.render()
+                         text_input = gr.Textbox(
+                            value="An impressionist painting",
+                            label="Text Prompt",
+                            info="Describe the style you want to apply to the image, do not include the description of the image content itself",
+                            lines=2,
+                            placeholder="Enter a text prompt",
+                        )
+                        image_input = gr.Image(
+                            label="Content image (will be resized to 512x512)",
+                            interactive=True,
+                        )
                         with gr.Accordion("Advanced settings"):
                             with gr.Column():
-                                cfg_slider.render()
-                                content_slider.render()
-                                style_slider.render()
-                                seed_slider.render()
+                                cfg_slider = gr.Slider(
+                                    0,
+                                    15,
+                                    value=7.5,
+                                    label="Classifier Free Guidance (CFG) Scale",
+                                    info="higher values give more style, 7.5 should be good for most cases",
+                                )
+                                content_slider = gr.Slider(
+                                    0,
+                                    9,
+                                    value=4,
+                                    step=1,
+                                    label="Number of content control layer",
+                                    info="higher values make it more similar to original image. Default to control first 4 layers",
+                                )
+                                style_slider = gr.Slider(
+                                    0,
+                                    9,
+                                    value=9,
+                                    step=1,
+                                    label="Number of style control layer",
+                                    info="higher values make it more similar to target style. Default to control first 9 layers, usually not necessary to change.",
+                                )
+                                seed_slider = gr.Slider(
+                                    0,
+                                    100,
+                                    value=0,
+                                    step=1,
+                                    label="Seed",
+                                    info="Random seed for the model",
+                                )
                         submit_btn = gr.Button("Submit")
                     with gr.Column():
-                        image_output.render()
+                        image_output= gr.Image(format="png")
                         gr.Examples(
                             examples = examples,
                             inputs = [text_input, image_input],