basso4 commited on
Commit
eb29fc5
Β·
verified Β·
1 Parent(s): a68557b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -188,7 +188,7 @@ if __name__ == "__main__":
188
 
189
  #mask
190
  mask_cv = mask_cv.convert("L").resize((384,512))
191
- mask_cv = np.array(mask_cv)
192
  mask_cv = 255-mask_cv
193
 
194
  #Normalize mask to binary values
 
188
 
189
  #mask
190
  mask_cv = mask_cv.convert("L").resize((384,512))
191
+ mask_cv = np.array(mask_cv).astype(np.uint8)
192
  mask_cv = 255-mask_cv
193
 
194
  #Normalize mask to binary values