22

app.py

@@ -125,7 +125,7 @@ def inference(
     scale_target = 2
     # rescale to prevent OOM
     input_image = edit_dict['background']
-    input_image, width, height = image_utils.rescale_image(input_image, 1, 8, max_size=1280)
+    input_image, width, height = image_utils.rescale_image(input_image, scale_target, 8, max_size=1280)
     sample_size = [height, width]
     
     print("DEBUG: inpaint_image")
@@ -136,14 +136,14 @@ def inference(
 
     print("DEBUG: mask_image")
     if mask_image is not None:
-        mask_image, w, h = image_utils.rescale_image(mask_image, 1, 8, max_size=1280)
+        mask_image, w, h = image_utils.rescale_image(mask_image, scale_target, 8, max_size=1280)
         mask_image = get_image_latent(mask_image, sample_size=sample_size)[:, :1, 0]
     else:
         mask_image = torch.ones([1, 1, sample_size[0], sample_size[1]]) * 255
 
     print("DEBUG: control_image_torch")
     processor = Processor('canny')
-    control_image, w, h = image_utils.rescale_image(input_image, 1, 8, max_size=1280)
+    control_image, w, h = image_utils.rescale_image(input_image, scale_target, 8, max_size=1280)
     control_image = control_image.resize((1024, 1024))
     control_image = processor(control_image, to_pil=True)
     control_image = control_image.resize((width, height))
@@ -198,7 +198,7 @@ with gr.Blocks() as demo:
                     brush= gr.Brush(
                         colors=["#FFFFFF"], 
                         color_mode="fixed",
-                        default_size=75
+                        # default_size=75
                     ),
                     label="Edit Image"
                 )