michaelapplydesign commited on
Commit
bc435a1
·
1 Parent(s): 842f4a8
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -33,9 +33,12 @@ def predict(input_img1,input_img2):
33
  # image = Image.open(requests.get("https://applydesignblobs-chh5aahjdzh0cnew.z01.azurefd.net/spaceimages/org_sqr_7fee0869-3187-4363-b5fb-5233e943649d.png", stream=True).raw)
34
  # mask = Image.open(requests.get("https://applydesign.blob.core.windows.net/spaceimages/mask_e85b1585-8.png", stream=True).raw)
35
 
 
 
 
36
  result_image = make_inpainting(positive_prompt='test1',
37
  image=image_to_byte_array(input_img1),
38
- mask_image=np.array(input_img2),
39
  negative_prompt="xxx",
40
  )
41
 
 
33
  # image = Image.open(requests.get("https://applydesignblobs-chh5aahjdzh0cnew.z01.azurefd.net/spaceimages/org_sqr_7fee0869-3187-4363-b5fb-5233e943649d.png", stream=True).raw)
34
  # mask = Image.open(requests.get("https://applydesign.blob.core.windows.net/spaceimages/mask_e85b1585-8.png", stream=True).raw)
35
 
36
+ canvas_mask = np.array(input_img2)
37
+ mask = get_mask(canvas_mask)
38
+
39
  result_image = make_inpainting(positive_prompt='test1',
40
  image=image_to_byte_array(input_img1),
41
+ mask_image=mask,
42
  negative_prompt="xxx",
43
  )
44