attention-refocusing commited on
Commit
5f3ed23
β€’
1 Parent(s): 270f5ec

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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'])