forgot to remove the hint/mask dict

app.py

@@ -34,7 +34,7 @@ if _model.failed != False:
 def generate(
         prompt = 'An elderly man having a great time in the park.',
         neg_prompt = '',
-        image = { 'image': None, 'mask': None },
+        image = None,
         inference_steps = 20,
         cfg = 12.0,
         seed = 0,
@@ -52,27 +52,17 @@ def generate(
     if seed < 0:
         seed = -seed
     inference_steps = int(inference_steps)
-    if image is not None:
-        hint_image = image['image']
-        mask_image = image['mask']
-    else:
-        hint_image = None
-        mask_image = None
+    hint_image = image
     if hint_image is not None:
         if hint_image.mode != 'RGB':
             hint_image = hint_image.convert('RGB')
         if hint_image.size != (width, height):
             hint_image = ImageOps.fit(hint_image, (width, height), method = Image.Resampling.LANCZOS)
-    if mask_image is not None:
-        if mask_image.mode != 'L':
-            mask_image = mask_image.convert('L')
-        if mask_image.size != (width, height):
-            mask_image = ImageOps.fit(mask_image, (width, height), method = Image.Resampling.LANCZOS)
     images = _model.generate(
             prompt = [prompt] * _model.device_count,
             neg_prompt = neg_prompt,
             hint_image = hint_image,
-            mask_image = mask_image,
+            mask_image = None,
             inference_steps = inference_steps,
             cfg = cfg,
             height = height,