Update app.py
Browse files
app.py
CHANGED
|
@@ -24,8 +24,9 @@ def perfect_remove_bg(img):
|
|
| 24 |
footprint=np.ones((3,3), np.uint8)
|
| 25 |
)
|
| 26 |
|
| 27 |
-
# U²Net
|
| 28 |
-
|
|
|
|
| 29 |
|
| 30 |
# Combine masks
|
| 31 |
final_mask = np.where(u2net_mask > 200, mask, u2net_mask)
|
|
|
|
| 24 |
footprint=np.ones((3,3), np.uint8)
|
| 25 |
)
|
| 26 |
|
| 27 |
+
# U²Net mask (FIXED - now has all parentheses closed)
|
| 28 |
+
u2net_result = remove(img, session=u2net_session)
|
| 29 |
+
u2net_mask = np.array(u2net_result.split()[-1])
|
| 30 |
|
| 31 |
# Combine masks
|
| 32 |
final_mask = np.where(u2net_mask > 200, mask, u2net_mask)
|