Spaces:
Runtime error
Runtime error
Ahsen Khaliq
commited on
Commit
·
4d2617e
1
Parent(s):
7422b24
Update app.py
Browse files
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(
|