hysts HF Staff commited on
Commit
5bdb8db
·
1 Parent(s): f33be2d

Fix type annotation

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -70,7 +70,7 @@ def binarize_mask(mask: np.ndarray) -> np.ndarray:
70
  return mask
71
 
72
 
73
- def update_trimap(foreground_mask_editor: gr.ImageEditor, unknown_mask_editor: gr.ImageEditor) -> np.ndarray:
74
  foreground = foreground_mask_editor["layers"][0]
75
  foreground = binarize_mask(foreground)
76
 
 
70
  return mask
71
 
72
 
73
+ def update_trimap(foreground_mask_editor: dict, unknown_mask_editor: dict) -> np.ndarray:
74
  foreground = foreground_mask_editor["layers"][0]
75
  foreground = binarize_mask(foreground)
76