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
|