Spaces:

yizhangliu
/

ImgCleaner

Running

yizhangliu commited on Nov 11, 2022

Commit

809a8a0

•

1 Parent(s): 5117bd8

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -233,7 +233,7 @@ def predict(dict):
     '''
     image = Image.fromarray(dict["image"])
-    mask = np.array(Image.fromarray(dict["mask"]).convert("RGB"))
     print(f'mask___1 = {mask.shape}')
     output = model_process(dict["image"], dict["mask"])

     '''
     image = Image.fromarray(dict["image"])
+    mask = np.array(Image.fromarray(dict["mask"]).convert("L"))
     print(f'mask___1 = {mask.shape}')
     output = model_process(dict["image"], dict["mask"])