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 = [
|