Image Guidance Scale, Strength and Black mask

app.py

@@ -30,6 +30,8 @@ def check(
     denoising_steps,
     num_inference_steps,
     guidance_scale,
+    image_guidance_scale,
+    strength,
     randomize_seed,
     seed,
     debug_mode,
@@ -72,6 +74,8 @@ def uncrop(
     denoising_steps,
     num_inference_steps,
     guidance_scale,
+    image_guidance_scale,
+    strength,
     randomize_seed,
     seed,
     debug_mode,
@@ -88,6 +92,8 @@ def uncrop(
         denoising_steps,
         num_inference_steps,
         guidance_scale,
+        image_guidance_scale,
+        strength,
         randomize_seed,
         seed,
         debug_mode
@@ -122,6 +128,12 @@ def uncrop(
     if guidance_scale is None:
         guidance_scale = 7
 
+    if image_guidance_scale is None:
+        image_guidance_scale = 1.5
+
+    if strength is None:
+        strength = 0.99
+
     if randomize_seed:
         seed = random.randint(0, max_64_bit_int)
 
@@ -182,7 +194,7 @@ def uncrop(
 
     # Mask
     mask_image = Image.new(mode = input_image.mode, size = (output_width, output_height), color = (255, 255, 255, 0))
-    black_mask = Image.new(mode = input_image.mode, size = (original_width - smooth_border, original_height - smooth_border), color = (127, 127, 127, 0))
+    black_mask = Image.new(mode = input_image.mode, size = (original_width - smooth_border, original_height - smooth_border), color = (0, 0, 0, 0))
     mask_image.paste(black_mask, (enlarge_left + (smooth_border // 2), enlarge_top + (smooth_border // 2)))
     mask_image = mask_image.filter(ImageFilter.BoxBlur((smooth_border // 2)))
 
@@ -226,6 +238,8 @@ def uncrop(
         mask_image = mask_image,
         num_inference_steps = num_inference_steps,
         guidance_scale = guidance_scale,
+        image_guidance_scale = image_guidance_scale,
+        strength = strength,
         denoising_steps = denoising_steps,
         show_progress_bar = True
     ).images[0]
@@ -308,17 +322,21 @@ with gr.Blocks() as interface:
     with gr.Row():
         prompt = gr.Textbox(label = 'Prompt', info = "Describe the subject, the background and the style of image; 77 token limit", placeholder = 'Describe what you want to see in the entire image')
     with gr.Row():
-         with gr.Accordion("Advanced options", open = False):
+        with gr.Accordion("Advanced options", open = False):
             negative_prompt = gr.Textbox(label = 'Negative prompt', placeholder = 'Describe what you do NOT want to see in the entire image', value = 'Border, frame, painting, scribbling, smear, noise, blur, watermark')
             smooth_border = gr.Slider(minimum = 0, maximum = 1024, value = 0, step = 2, label = "Smooth border", info = "lower=preserve original, higher=seamless")
             denoising_steps = gr.Slider(minimum = 0, maximum = 1000, value = 1000, step = 1, label = "Denoising", info = "lower=irrelevant result, higher=relevant result")
             num_inference_steps = gr.Slider(minimum = 10, maximum = 100, value = 50, step = 1, label = "Number of inference steps", info = "lower=faster, higher=image quality")
             guidance_scale = gr.Slider(minimum = 1, maximum = 13, value = 7, step = 0.1, label = "Classifier-Free Guidance Scale", info = "lower=image quality, higher=follow the prompt")
+            image_guidance_scale = gr.Slider(minimum = 1, value = 1.5, step = 0.1, label = "Image Guidance Scale", info = "lower=image quality, higher=follow the image")
+            strength = gr.Number(value = 0.99, minimum = 0.01, maximum = 1.0, step = 0.01, label = "Strength", info = "lower=follow the original area (discouraged), higher=redraw from scratch")
             randomize_seed = gr.Checkbox(label = "\U0001F3B2 Randomize seed (not working, always checked)", value = True, info = "If checked, result is always different")
             seed = gr.Slider(minimum = 0, maximum = max_64_bit_int, step = 1, randomize = True, label = "Seed (if not randomized)")
             debug_mode = gr.Checkbox(label = "Debug mode", value = False, info = "Show intermediate results")
+
     with gr.Row():
         submit = gr.Button("Uncrop", variant = "primary")
+
     with gr.Row():
         uncropped_image = gr.Image(label = "Uncropped image")
     with gr.Row():
@@ -346,6 +364,8 @@ with gr.Blocks() as interface:
         denoising_steps,
         num_inference_steps,
         guidance_scale,
+        image_guidance_scale,
+        strength,
         randomize_seed,
         seed,
         debug_mode
@@ -362,6 +382,8 @@ with gr.Blocks() as interface:
         denoising_steps,
         num_inference_steps,
         guidance_scale,
+        image_guidance_scale,
+        strength,
         randomize_seed,
         seed,
         debug_mode
@@ -386,6 +408,8 @@ with gr.Blocks() as interface:
             denoising_steps,
             num_inference_steps,
             guidance_scale,
+            image_guidance_scale,
+            strength,
             randomize_seed,
             seed,
             debug_mode
@@ -410,6 +434,8 @@ with gr.Blocks() as interface:
                     1000,
                     50,
                     7,
+                    1.5,
+                    0.99,
                     True,
                     42,
                     False
@@ -426,6 +452,8 @@ with gr.Blocks() as interface:
                     1000,
                     50,
                     7,
+                    1.5,
+                    0.99,
                     True,
                     42,
                     False
@@ -442,6 +470,8 @@ with gr.Blocks() as interface:
                     1000,
                     50,
                     7,
+                    1.5,
+                    0.99,
                     True,
                     42,
                     False