Omnibus commited on
Commit
51cca02
Β·
1 Parent(s): ba0ae7d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -71,7 +71,7 @@ def get_seg(image, object):
71
  mask_array_three_channel_invert = 1-mask_array_three_channel
72
  #blur_image_reverse_mask = blur_image*mask_array_three_channel_invert
73
  seg_box=[]
74
- for i in range(pred):
75
  object_number = int(object.split('i')[0])
76
  mask_array = np.asarray(pred[object_number]['mask'])/255
77
  image_array = np.asarray(image)
 
71
  mask_array_three_channel_invert = 1-mask_array_three_channel
72
  #blur_image_reverse_mask = blur_image*mask_array_three_channel_invert
73
  seg_box=[]
74
+ for i in range(len(pred)):
75
  object_number = int(object.split('i')[0])
76
  mask_array = np.asarray(pred[object_number]['mask'])/255
77
  image_array = np.asarray(image)