Update injection_main_HF.py

injection_main_HF.py

@@ -606,6 +606,10 @@ if __name__ == "__main__":
                 """)
                 with gr.Row():
                     with gr.Column():
+                        image_input = gr.Image(
+                            label="Content image (will be resized to 512x512)",
+                            interactive=True,
+                        )
                         text_input = gr.Textbox(
                             value="An impressionist painting",
                             label="Text Prompt",
@@ -613,10 +617,6 @@ if __name__ == "__main__":
                             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 = gr.Slider(
@@ -656,7 +656,8 @@ if __name__ == "__main__":
                         gr.Examples(
                             examples = examples,
                             inputs = [text_input, image_input],
-                            cache_examples=False
+                            outputs = [image_output]
+                            cache_examples='lazy'
                         )
             submit_btn.click(
                 fn=style_transfer_app,