Spaces:
Running on T4
Running on T4
Commit ·
cdc3c72
1
Parent(s): 3db9095
Fix mask_editor_data background extraction
Browse files
app.py
CHANGED
|
@@ -414,8 +414,10 @@ def gradio_estimate(
|
|
| 414 |
layers = mask_editor_data.get("layers")
|
| 415 |
if isinstance(layers, list) and layers:
|
| 416 |
editor_mask = layers[-1]
|
| 417 |
-
|
| 418 |
editor_mask = mask_editor_data.get("composite")
|
|
|
|
|
|
|
| 419 |
else:
|
| 420 |
editor_mask = mask_editor_data
|
| 421 |
|
|
|
|
| 414 |
layers = mask_editor_data.get("layers")
|
| 415 |
if isinstance(layers, list) and layers:
|
| 416 |
editor_mask = layers[-1]
|
| 417 |
+
elif mask_editor_data.get("composite"):
|
| 418 |
editor_mask = mask_editor_data.get("composite")
|
| 419 |
+
else:
|
| 420 |
+
editor_mask = mask_editor_data.get("background")
|
| 421 |
else:
|
| 422 |
editor_mask = mask_editor_data
|
| 423 |
|