Tobias Cornille commited on
Commit
73f96dd
1 Parent(s): 77202d8

Force use latest gradio

Browse files
Files changed (2) hide show
  1. app.py +3 -1
  2. 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 = zip(panoptic_bool_masks, category_names)
 
 
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