Spaces:
Runtime error
Runtime error
attention-refocusing
commited on
Commit
β’
5f3ed23
1
Parent(s):
270f5ec
Update app.py
Browse files
app.py
CHANGED
@@ -150,6 +150,7 @@ class ImageMask(gr.components.Image):
|
|
150 |
mask = self.postprocess(mask)
|
151 |
x = {'image': x, 'mask': mask}
|
152 |
print('vao preprocess-------------------------')
|
|
|
153 |
if (hh['image'].min()!=255) and (hh['mask'][:,:,:3].max()==0):
|
154 |
|
155 |
hh['mask'] = binarize_2(hh['image'])
|
|
|
150 |
mask = self.postprocess(mask)
|
151 |
x = {'image': x, 'mask': mask}
|
152 |
print('vao preprocess-------------------------')
|
153 |
+
hh = super().preprocess(x)
|
154 |
if (hh['image'].min()!=255) and (hh['mask'][:,:,:3].max()==0):
|
155 |
|
156 |
hh['mask'] = binarize_2(hh['image'])
|