Spaces:
Sleeping
Sleeping
Ahsen Khaliq
commited on
Commit
•
b179895
1
Parent(s):
353b3be
Update app.py
Browse files
app.py
CHANGED
@@ -11,9 +11,8 @@ os.mkdir("data")
|
|
11 |
os.mkdir("dataout")
|
12 |
model = hub.Module(name='U2Net')
|
13 |
def infer(img,mask,option):
|
14 |
-
img =
|
15 |
-
|
16 |
-
mask = mask.resize((600,600))
|
17 |
img.save("./data/data.png")
|
18 |
if option == "automatic (U2net)":
|
19 |
result = model.Segmentation(
|
|
|
11 |
os.mkdir("dataout")
|
12 |
model = hub.Module(name='U2Net')
|
13 |
def infer(img,mask,option):
|
14 |
+
img = ImageOps.contain(img, (700,700))
|
15 |
+
mask = ImageOps.contain(mask, (700,700))
|
|
|
16 |
img.save("./data/data.png")
|
17 |
if option == "automatic (U2net)":
|
18 |
result = model.Segmentation(
|