Update app.py

app.py

@@ -111,13 +111,10 @@ def predict(jpg, msk):
 
 
 def infer(img,option):
-  print(type(img))
-  print(type(img["image"]))
-  print(type(img["mask"]))
-  imageio.imwrite("./data/data.png", img["image"])
-  if option == "automatic (U2net)":
+  imageio.imwrite("./data/data.png", img)
+  if option == "automatic":
       result = model.Segmentation(
-          images=[cv2.cvtColor(img["image"], cv2.COLOR_RGB2BGR)],
+          images=[cv2.cvtColor(img, cv2.COLOR_RGB2BGR)],
           paths=None,
           batch_size=1,
           input_size=320,