Spaces:
Running
Running
yizhangliu
commited on
Commit
·
e9e2ff1
1
Parent(s):
59dac77
Update app.py
Browse files
app.py
CHANGED
@@ -104,7 +104,7 @@ def preprocess_mask(mask):
|
|
104 |
mask = torch.from_numpy(mask)
|
105 |
return mask
|
106 |
|
107 |
-
def
|
108 |
# alpha_channel = None
|
109 |
# nparr = np.frombuffer(img_bytes, np.uint8)
|
110 |
if gray:
|
|
|
104 |
mask = torch.from_numpy(mask)
|
105 |
return mask
|
106 |
|
107 |
+
def load_img_1_(nparr, gray: bool = False):
|
108 |
# alpha_channel = None
|
109 |
# nparr = np.frombuffer(img_bytes, np.uint8)
|
110 |
if gray:
|