Update app.py

app.py

@@ -11,10 +11,9 @@ os.mkdir("data")
 os.mkdir("dataout")
 model = hub.Module(name='U2Net')
 def infer(img,mask,option):
-  basewidth = 600
-  wpercent = (basewidth/float(img.size[0]))
-  hsize = int((float(img.size[1])*float(wpercent)))
-  img = img.resize((basewidth,hsize), Image.ANTIALIAS)
+  img = img.resize((600,600))   # image resizing
+    
+  mask = mask.resize((600,600))
   img.save("./data/data.png")
   if option == "automatic (U2net)":
       result = model.Segmentation(