Spaces:
Runtime error
Runtime error
Tobias Cornille
commited on
Commit
·
391271a
1
Parent(s):
37f9596
Fix category names
Browse files
app.py
CHANGED
|
@@ -315,6 +315,7 @@ def generate_panoptic_mask(
|
|
| 315 |
]
|
| 316 |
stuff_category_names = [category["name"] for category in stuff_categories]
|
| 317 |
thing_category_names = [category["name"] for category in thing_categories]
|
|
|
|
| 318 |
else:
|
| 319 |
# parse inputs
|
| 320 |
thing_category_names = [
|
|
|
|
| 315 |
]
|
| 316 |
stuff_category_names = [category["name"] for category in stuff_categories]
|
| 317 |
thing_category_names = [category["name"] for category in thing_categories]
|
| 318 |
+
category_names = thing_category_names + stuff_category_names
|
| 319 |
else:
|
| 320 |
# parse inputs
|
| 321 |
thing_category_names = [
|