fix mask

app.py

@@ -79,21 +79,21 @@ def process_images(image, image2=None,prompt="a girl",inpaint_model="black-fores
         generators.append(generator)
         
         width,height = convert_to_fit_size(image.size)
-        print(f"fit {width}x{height}")
+        #print(f"fit {width}x{height}")
         width,height = adjust_to_multiple_of_32(width,height)
-        print(f"multiple {width}x{height}")
+        #print(f"multiple {width}x{height}")
         image = image.resize((width, height), Image.LANCZOS)
         mask_image = mask_image.resize((width, height), Image.NEAREST)
 
         output = pipe(prompt=prompt, image=image, mask_image=mask_image,generator=generator,strength=strength,width=width,height=height,
                       guidance_scale=0,num_inference_steps=num_inference_steps,max_sequence_length=256)
 
-        return output.images[0]
+        return output.images[0],mask_image
 
 
-    output = process_inpaint(image["background"],mask,prompt,inpaint_model,strength,seed)
+    output,mask_image = process_inpaint(image["background"],mask,prompt,inpaint_model,strength,seed)
         
-    return output,mask
+    return output,mask_image
     
 
 def read_file(path: str) -> str: