Spaces:
Runtime error
Runtime error
Tobias Cornille
commited on
Commit
·
73f96dd
1
Parent(s):
77202d8
Force use latest gradio
Browse files- app.py +3 -1
- requirements.txt +1 -1
app.py
CHANGED
|
@@ -353,7 +353,9 @@ def generate_panoptic_mask(
|
|
| 353 |
.astype(int)
|
| 354 |
)
|
| 355 |
category_names = ["background"] + stuff_category_names + thing_category_names
|
| 356 |
-
subsection_label_pairs =
|
|
|
|
|
|
|
| 357 |
|
| 358 |
return (image_array, subsection_label_pairs)
|
| 359 |
|
|
|
|
| 353 |
.astype(int)
|
| 354 |
)
|
| 355 |
category_names = ["background"] + stuff_category_names + thing_category_names
|
| 356 |
+
subsection_label_pairs = [
|
| 357 |
+
(panoptic_bool_masks[i], category_names[i]) for i in range(len(category_names))
|
| 358 |
+
]
|
| 359 |
|
| 360 |
return (image_array, subsection_label_pairs)
|
| 361 |
|
requirements.txt
CHANGED
|
@@ -1,7 +1,7 @@
|
|
| 1 |
accelerate
|
| 2 |
addict
|
| 3 |
diffusers
|
| 4 |
-
gradio
|
| 5 |
huggingface_hub
|
| 6 |
matplotlib
|
| 7 |
numpy
|
|
|
|
| 1 |
accelerate
|
| 2 |
addict
|
| 3 |
diffusers
|
| 4 |
+
gradio==3.27.0
|
| 5 |
huggingface_hub
|
| 6 |
matplotlib
|
| 7 |
numpy
|