Ahsen Khaliq commited on
Commit
4d2617e
·
1 Parent(s): 7422b24

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ model = hub.Module(name='U2Net')
13
  def infer(img,mask,option):
14
  img = ImageOps.contain(img, (700,700))
15
  width, height = img.size
16
- mask = mask.resize(width,height)
17
  img.save("./data/data.png")
18
  if option == "automatic (U2net)":
19
  result = model.Segmentation(
 
13
  def infer(img,mask,option):
14
  img = ImageOps.contain(img, (700,700))
15
  width, height = img.size
16
+ mask = mask.resize((width,height))
17
  img.save("./data/data.png")
18
  if option == "automatic (U2net)":
19
  result = model.Segmentation(