update

app.py

@@ -1,6 +1,6 @@
 import huggingface_hub
 import gradio as gr
-from stable_diffusion_reference_only.pipelines.stable_diffusion_reference_only_pipeline import (
+from stable_diffusion_reference_only.pipelines.pipeline_stable_diffusion_reference_only import (
     StableDiffusionReferenceOnlyPipeline,
 )
 from anime_segmentation import get_model as get_anime_segmentation_model
@@ -29,6 +29,7 @@ segment_model = get_anime_segmentation_model(
     model_path=huggingface_hub.hf_hub_download("skytnt/anime-seg", "isnetis.ckpt")
 ).to(device)
 
+
 def character_segment(img):
     if img is None:
         return None
@@ -36,6 +37,7 @@ def character_segment(img):
     img = cv2.cvtColor(img, cv2.COLOR_RGBA2RGB)
     return img
 
+
 def color_inversion(img):
     if img is None:
         return None
@@ -85,6 +87,13 @@ def style_transfer(prompt, blueprint, num_inference_steps):
     blueprint = get_line_art(blueprint)
     blueprint = color_inversion(blueprint)
     return inference(prompt, blueprint, num_inference_steps)
+
+
+def resize(img, new_height, new_width):
+    img = Image.fromarray(img).resize((int(new_width), int(new_height)), Image.BILINEAR)
+    return np.array(img)
+
+
 with gr.Blocks() as demo:
     gr.Markdown(
         """
@@ -94,7 +103,21 @@ with gr.Blocks() as demo:
     )
     with gr.Row():
         with gr.Column():
-            prompt_input_compoent = gr.Image(shape=(512, 512), label="prompt")
+            prompt_input_compoent = gr.Image(label="prompt")
+            with gr.Row():
+                prompt_new_height = gr.Number(512, label="height", minimum=1)
+                prompt_new_width = gr.Number(512, label="width", minimum=1)
+                prompt_resize_button = gr.Button("prompt resize")
+                prompt_resize_button.click(
+                    resize,
+                    inputs=[
+                        prompt_input_compoent,
+                        prompt_new_height,
+                        prompt_new_width,
+                    ],
+                    outputs=prompt_input_compoent,
+                )
+
             prompt_character_segment_button = gr.Button(
                 "character segment",
             )
@@ -104,7 +127,22 @@ with gr.Blocks() as demo:
                 outputs=prompt_input_compoent,
             )
         with gr.Column():
-            blueprint_input_compoent = gr.Image(shape=(512, 512), label="blueprint")
+            blueprint_input_compoent = gr.Image(label="blueprint")
+
+            with gr.Row():
+                blueprint_new_height = gr.Number(512, label="height", minimum=1)
+                blueprint_new_width = gr.Number(512, label="width", minimum=1)
+                blueprint_resize_button = gr.Button("blueprint resize")
+                blueprint_resize_button.click(
+                    resize,
+                    inputs=[
+                        blueprint_input_compoent,
+                        blueprint_new_height,
+                        blueprint_new_width,
+                    ],
+                    outputs=blueprint_input_compoent,
+                )
+
             blueprint_character_segment_button = gr.Button("character segment")
             blueprint_character_segment_button.click(
                 character_segment,
@@ -128,7 +166,7 @@ with gr.Blocks() as demo:
                 outputs=blueprint_input_compoent,
             )
         with gr.Column():
-            result_output_component = gr.Image(shape=(512, 512), label="result")
+            result_output_component = gr.Image(label="result")
             num_inference_steps_input_component = gr.Number(
                 20, label="num inference steps", minimum=1, maximum=1000, step=1
             )
@@ -179,5 +217,6 @@ with gr.Blocks() as demo:
             fn=lambda x, y: None,
             cache_examples=True,
         )
+
 if __name__ == "__main__":
     demo.queue(max_size=5).launch()