Spaces:

segments
/

panoptic-segment-anything

Runtime error

Tobias Cornille commited on Apr 18, 2023

Commit

77a096a

•

1 Parent(s): 94040eb

Fix bitmap file output

Files changed (1) hide show

app.py CHANGED Viewed

@@ -267,7 +267,7 @@ def inds_to_segments_format(
 ):
     panoptic_inds_array = panoptic_inds.numpy().astype(np.uint32)
     bitmap_file = bitmap2file(panoptic_inds_array, is_segmentation_bitmap=True)
-    output_file.write(bitmap_file)
     unique_inds = np.unique(panoptic_inds_array)
     stuff_annotations = [

 ):
     panoptic_inds_array = panoptic_inds.numpy().astype(np.uint32)
     bitmap_file = bitmap2file(panoptic_inds_array, is_segmentation_bitmap=True)
+    output_file.write(bitmap_file.read())
     unique_inds = np.unique(panoptic_inds_array)
     stuff_annotations = [