Code Improvement.

app.py

@@ -3,10 +3,8 @@ from webui import (
     create_interface_texture_synthesis,
     create_interface_style_t2i,
     create_interface_style_transfer,
-    Runner
 )
-
-
+from webui.runner import Runner
 
 
 def main():

webui/__init__.py

@@ -2,4 +2,4 @@
 from .tab_style_t2i import create_interface_style_t2i
 from .tab_style_transfer import create_interface_style_transfer
 from .tab_texture_synthesis import create_interface_texture_synthesis
-from .runner import Runner
+# from .runner import Runner

webui/runner.py

@@ -2,7 +2,7 @@ import torch
 from PIL import Image
 from diffusers import DDIMScheduler
 from accelerate.utils import set_seed
-from torchvision.transforms.functional import to_pil_image, to_tensor
+from torchvision.transforms.functional import to_pil_image, to_tensor, resize
 
 from pipeline_sd import ADPipeline
 from pipeline_sdxl import ADPipeline as ADXLPipeline
@@ -30,6 +30,8 @@ class Runner:
             self.sd15.classifier = self.sd15.unet
 
     def preprocecss(self, image: Image.Image, height=None, width=None):
+        image = resize(image, size=512)
+        
         if width is None or height is None: 
             width, height = image.size
         new_width = (width // 64) * 64

webui/tab_style_transfer.py

@@ -13,10 +13,8 @@ def create_interface_style_transfer(runner):
         with gr.Row():
             with gr.Column():
                 with gr.Row():
-                    content_image = gr.Image(label='Input Content Image', type='pil', interactive=True,
-                                             value=Image.open('examples/c1.jpg').convert('RGB') if os.path.exists('examples/c1.jpg') else None)
-                    style_image = gr.Image(label='Input Style Image', type='pil', interactive=True,
-                                           value=Image.open('examples/s1.jpg').convert('RGB') if os.path.exists('examples/s1.jpg') else None)
+                    content_image = gr.Image(label='Input Content Image', type='pil', interactive=True, value=None)
+                    style_image = gr.Image(label='Input Style Image', type='pil', interactive=True, value=None)
 
                 run_button = gr.Button(value='Run')
 
@@ -33,9 +31,23 @@ def create_interface_style_transfer(runner):
             with gr.Column():
                 gr.Markdown('#### Output Image:\n')
                 result_gallery = gr.Gallery(label='Output', elem_id='gallery', columns=2, height='auto', preview=True)
-                
+                gr.Markdown(
+                    'Notes:\n'
+                    '* If you find the style effect insufficient, you can try increasing the `Number of Steps` or decreasing the `Content Weight`\n'
+                    '* For face stylization, we generally recommend using a `Content Weight` of `0.26` for most faces.'
+                )
                 gr.Examples(
-                    [[Image.open('./webui/images/lecun.png').convert('RGB'), Image.open('./webui/images/40.jpg').convert('RGB'), 300, 0.23]],
+                    [
+                        [Image.open('./webui/images/deer1.jpg').convert('RGB'), Image.open('./webui/images/35.jpg').convert('RGB'), 200, 0.25],
+                        [Image.open('./webui/images/4.jpg').convert('RGB'), Image.open('./webui/images/17.jpg').convert('RGB'), 200, 0.25],
+                        [Image.open('./webui/images/18.jpg').convert('RGB'), Image.open('./webui/images/3.jpg').convert('RGB'), 200, 0.25],
+                        [Image.open('./webui/images/8.jpg').convert('RGB'), Image.open('./webui/images/sketch.png').convert('RGB'), 300, 0.2],
+                        [Image.open('./webui/images/5.png').convert('RGB'), Image.open('./webui/images/45.jpg').convert('RGB'), 200, 0.25],
+                        [Image.open('./webui/images/9.jpg').convert('RGB'), Image.open('./webui/images/16.jpg').convert('RGB'), 200, 0.25],
+                        [Image.open('./webui/images/7.png').convert('RGB'), Image.open('./webui/images/31.jpg').convert('RGB'), 200, 0.25],
+                        [Image.open('./webui/images/14.jpg').convert('RGB'), Image.open('./webui/images/48.jpg').convert('RGB'), 200, 0.25],
+                        [Image.open('./webui/images/lecun.png').convert('RGB'), Image.open('./webui/images/40.jpg').convert('RGB'), 300, 0.23],
+                    ],
                     [content_image, style_image, num_steps, content_weight]
                 )