add example

.gitattributes

@@ -34,5 +34,3 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
 diffsynth/tokenizer_configs/kolors/tokenizer/vocab.txt filter=lfs diff=lfs merge=lfs -text
-images/image1.png filter=lfs diff=lfs merge=lfs -text
-images/image2.png filter=lfs diff=lfs merge=lfs -text

app.py

@@ -154,23 +154,12 @@ with gr.Blocks() as app:
             
             with gr.Accordion(label="Image"):
                 num_inference_steps = gr.Slider(minimum=1, maximum=100, value=20, step=1, interactive=True, label="Inference steps")
-                height = gr.Slider(minimum=1024, maximum=1024, value=1024, step=64, interactive=False, label="Height", visible=False)
-                width = gr.Slider(minimum=1024, maximum=1024, value=1024, step=64, interactive=False, label="Width", visible=False)
+                height = gr.Slider(minimum=64, maximum=2048, value=1024, step=64, interactive=True, label="Height")
+                width = gr.Slider(minimum=64, maximum=2048, value=1024, step=64, interactive=True, label="Width")
                 with gr.Column():
                     use_fixed_seed = gr.Checkbox(value=True, interactive=False, label="Use fixed seed")
                     seed = gr.Number(minimum=0, maximum=10**9, value=0, interactive=True, label="Random seed", show_label=False)
 
-            with gr.Row(elem_id="pro-tips"):
-                gr.Markdown("""
-                # Usage: 
-                
-                1. Enter a prompt and click "Generate" to create an image.
-                2. Click "Set as painter's background" to use the generated image as the canvas.
-                3. In the painter, draw the area you want to repaint and set a local prompt. For multiple areas with different prompts, use layer1 and layer2, and click "Enable this layer" to activate the canvas.
-                4. Click "Generate" again to obtain the repainted image.    
-                5. If you want to regenerate a new image as the background, uncheck "Enable this layer," update the prompt, and click "Generate."
-                """)
-
             @gr.on(
                 inputs=[model_type, model_path, prompt, negative_prompt, cfg_scale, embedded_guidance, num_inference_steps, height, width],
                 outputs=[prompt, negative_prompt, cfg_scale, embedded_guidance, num_inference_steps, height, width],
@@ -194,17 +183,10 @@ with gr.Blocks() as app:
                 canvas_list = []
                 for painter_layer_id in range(config["max_num_painter_layers"]):
                     with gr.Tab(label=f"Layer {painter_layer_id}"):
-                        enable_local_prompt_value = painter_layer_id == 0
-                        enable_local_prompt = gr.Checkbox(
-                            label="Enable this layer", 
-                            value=enable_local_prompt_value, 
-                            key=f"enable_local_prompt_{painter_layer_id}"
-                        )
-                        
+                        enable_local_prompt = gr.Checkbox(label="Enable", value=False, key=f"enable_local_prompt_{painter_layer_id}")
                         local_prompt = gr.Textbox(label="Local prompt", key=f"local_prompt_{painter_layer_id}")
-                        mask_scale = gr.Slider(minimum=0.0, maximum=5.0, value=1.0, step=0.1, interactive=True, label="Mask scale", key=f"mask_scale_{painter_layer_id}")
-                        canvas_size = (1024, 1024) if painter_layer_id == 0 else (1024, 1)
-                        canvas = gr.ImageEditor(canvas_size=canvas_size, sources=None, layers=False, interactive=True, image_mode="RGBA",
+                        mask_scale = gr.Slider(minimum=0.0, maximum=5.0, value=5.0, step=0.1, interactive=True, label="Mask scale", key=f"mask_scale_{painter_layer_id}")
+                        canvas = gr.ImageEditor(canvas_size=(512, 1), sources=None, layers=False, interactive=True, image_mode="RGBA",
                                                 brush=gr.Brush(default_size=100, default_color="#000000", colors=["#000000"]),
                                                 label="Painter", key=f"canvas_{painter_layer_id}")
                         @gr.on(inputs=[height, width, canvas], outputs=canvas, triggers=[height.change, width.change, canvas.clear, enable_local_prompt.change], show_progress="hidden")
@@ -273,32 +255,4 @@ with gr.Blocks() as app:
                         h, w = canvas["background"].shape[:2]
                         canvas["background"] = output_image.resize((w, h))
                     return tuple(canvas_list)
-                
-    canvas1 = {
-    "background": Image.open("images/image1.png"),
-    "layers": [np.array(Image.open("images/image1_layer.png"))],
-    "composite": "images/image1_layer.png",
-    }
-    canvas2 = {
-    "background": Image.open("images/image2.png"),
-    "layers": [np.array(Image.open("images/image2_layer.png"))],
-    "composite": "images/image2_layer.png",
-    }
-    canvas3 = {
-    "background": Image.open("images/image3.png"),
-    "layers": [np.array(Image.open("images/image3_layer.png"))],
-    "composite": "images/image3_layer.png",
-    }
-
-    print(*enable_local_prompt_list, *local_prompt_list, *mask_scale_list, *canvas_list)
-    with gr.Row():
-        show_case = gr.Examples(
-        examples=[
-            ["a girl", 0, "images/image1.png", True, "red hat", 3.0, canvas1],
-            ["an orange cat", 0, "images/image2.png", True, "a big crown on the cat", 3.0, canvas2],
-            ["A young man is riding a horse", 0, "images/image3.png", True, "A robot is riding a horse", 3.0, canvas3],
-        ],
-        inputs=[prompt, seed, output_image, enable_local_prompt_list[0], local_prompt_list[0], mask_scale_list[0], canvas_list[0]],
-        label=None
-        )
 app.launch()