diff --git a/language/en.json b/language/en.json
index fc241fc2..af1e614f 100644
--- a/language/en.json
+++ b/language/en.json
@@ -36,6 +36,15 @@
     "Top": "Top",
     "Bottom": "Bottom",
     "* \"Inpaint or Outpaint\" is powered by the sampler \"DPMPP Fooocus Seamless 2M SDE Karras Inpaint Sampler\" (beta)": "* \"Inpaint or Outpaint\" is powered by the sampler \"DPMPP Fooocus Seamless 2M SDE Karras Inpaint Sampler\" (beta)",
+    "Mask generation model": "Mask generation model",
+    "Cloth category": "Cloth category",
+    "Segmentation prompt": "Segmentation prompt",
+    "Advanced options": "Advanced options",
+    "SAM model": "SAM model",
+    "Quantization": "Quantization",
+    "Box Threshold": "Box Threshold",
+    "Text Threshold": "Text Threshold",
+    "Generate mask from image": "Generate mask from image",
     "Setting": "Setting",
     "Style": "Style",
     "Performance": "Performance",
@@ -376,15 +385,6 @@
     "Fooocus Enhance": "Fooocus Enhance",
     "Fooocus Cinematic": "Fooocus Cinematic",
     "Fooocus Sharp": "Fooocus Sharp",
-    "Mask generation model": "Mask generation model",
-    "Cloth category": "Cloth category",
-    "Segmentation prompt": "Segmentation prompt",
-    "Advanced options": "Advanced options",
-    "SAM model": "SAM model",
-    "Quantization": "Quantization",
-    "Box Threshold": "Box Threshold",
-    "Text Threshold": "Text Threshold",
-    "Generate mask from image": "Generate mask from image"
     "Drag any image generated by Fooocus here": "Drag any image generated by Fooocus here",
     "Metadata": "Metadata",
     "Apply Metadata": "Apply Metadata",
@@ -392,4 +392,4 @@
     "Image Prompt parameters are not included. Use a1111 for compatibility with Civitai.": "Image Prompt parameters are not included. Use a1111 for compatibility with Civitai.",
     "fooocus (json)": "fooocus (json)",
     "a1111 (plain text)": "a1111 (plain text)"
-}
+}
\ No newline at end of file
diff --git a/launch.py b/launch.py
index 0c101510..d8ea16ea 100644
--- a/launch.py
+++ b/launch.py
@@ -22,6 +22,7 @@ import fooocus_version
 from build_launcher import build_launcher
 from modules.launch_util import is_installed, run, python, run_pip, requirements_met
 from modules.model_loader import load_file_from_url
+from modules import config
 
 REINSTALL_ALL = False
 TRY_INSTALL_XFORMERS = False
diff --git a/modules/async_worker.py b/modules/async_worker.py
index c7511792..845d0b12 100644
--- a/modules/async_worker.py
+++ b/modules/async_worker.py
@@ -354,13 +354,17 @@ def worker():
                 inpaint_mask = inpaint_input_image['mask'][:, :, 0]
 
                 if inpaint_mask_upload_checkbox:
-                    if isinstance(inpaint_mask_image_upload, np.ndarray):
-                        if inpaint_mask_image_upload.ndim == 3:
-                            H, W, C = inpaint_image.shape
-                            inpaint_mask_image_upload = resample_image(inpaint_mask_image_upload, width=W, height=H)
-                            inpaint_mask_image_upload = np.mean(inpaint_mask_image_upload, axis=2)
-                            inpaint_mask_image_upload = (inpaint_mask_image_upload > 127).astype(np.uint8) * 255
-                            inpaint_mask = np.maximum(inpaint_mask, inpaint_mask_image_upload)
+                    if isinstance(inpaint_mask_image_upload, dict):
+                        if (isinstance(inpaint_mask_image_upload['image'], np.ndarray)
+                                and isinstance(inpaint_mask_image_upload['mask'], np.ndarray)
+                                and inpaint_mask_image_upload['image'].ndim == 3):
+                            inpaint_mask_image_upload = np.maximum(inpaint_mask_image_upload['image'], inpaint_mask_image_upload['mask'])
+                    if isinstance(inpaint_mask_image_upload, np.ndarray) and inpaint_mask_image_upload.ndim == 3:
+                        H, W, C = inpaint_image.shape
+                        inpaint_mask_image_upload = resample_image(inpaint_mask_image_upload, width=W, height=H)
+                        inpaint_mask_image_upload = np.mean(inpaint_mask_image_upload, axis=2)
+                        inpaint_mask_image_upload = (inpaint_mask_image_upload > 127).astype(np.uint8) * 255
+                        inpaint_mask = np.maximum(inpaint_mask, inpaint_mask_image_upload)
 
                 if int(inpaint_erode_or_dilate) != 0:
                     inpaint_mask = erode_or_dilate(inpaint_mask, inpaint_erode_or_dilate)
diff --git a/webui.py b/webui.py
index 67d84650..d3622049 100644
--- a/webui.py
+++ b/webui.py
@@ -211,8 +211,7 @@ with shared.gradio_root:
                                 example_inpaint_prompts.click(lambda x: x[0], inputs=example_inpaint_prompts, outputs=inpaint_additional_prompt, show_progress=False, queue=False)
 
                             with gr.Column(visible=False) as inpaint_mask_generation_col:
-                                inpaint_mask_image = grh.Image(label='Mask Upload', source='upload', type='numpy',
-                                                               height=500)
+                                inpaint_mask_image = grh.Image(label='Mask Upload', source='upload', type='numpy', tool='sketch', height=500, brush_color="#FFFFFF", mask_opacity=1)
                                 inpaint_mask_model = gr.Dropdown(label='Mask generation model',
                                                                  choices=flags.inpaint_mask_models,
                                                                  value=modules.config.default_inpaint_mask_model)