mask

app.py

@@ -101,7 +101,8 @@ def predict1(dict, prompt, num_samples):
 
 def predict2(dict, prompt, num_samples):
     dict['image'].save('image.png')
-    dict['mask'].save('mask.png')
+    if 'mask' in dict:
+        dict['mask'].save('mask.png')
     image = image_inpainting(prompt, num_samples=num_samples, image_path='image.png', mask_image_path='mask.png', is_origin=True)
     return image