Hugging Face's logo

Spaces:
hysts
/
ViTMatte
Runtime error

App Files Community
2
hysts HF staff commited on
Commit
11433a5
·
1 Parent(s): 85f6a01

Update

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -32,10 +32,10 @@ def binarize_mask(mask: np.ndarray) -> np.ndarray:
32
 
33
 
34
  def update_trimap(foreground_mask: dict[str, np.ndarray], unknown_mask: dict[str, np.ndarray]) -> np.ndarray:
35
- foreground = foreground_mask["mask"]
36
  foreground = binarize_mask(foreground)
37
 
38
- unknown = unknown_mask["mask"]
39
  unknown = binarize_mask(unknown)
40
 
41
  trimap = np.zeros_like(foreground)
 
32
 
33
 
34
  def update_trimap(foreground_mask: dict[str, np.ndarray], unknown_mask: dict[str, np.ndarray]) -> np.ndarray:
35
+ foreground = foreground_mask["mask"][:, :, 0]
36
  foreground = binarize_mask(foreground)
37
 
38
+ unknown = unknown_mask["mask"][:, :, 0]
39
  unknown = binarize_mask(unknown)
40
 
41
  trimap = np.zeros_like(foreground)