init

app.py

@@ -95,7 +95,7 @@ examples = [
 def predict1(dict, prompt):
     dict['image'].save('image.png')
     # dict['mask'].save('mask.png')
-    mediapipe_segmentation('image.png')
+    mediapipe_segmentation('image.png', 'm_mask.png')
     image = image_inpainting(prompt, image_path='image.png', mask_image_path='m_mask.png', is_origin=False)
     return image