Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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)
|