Update app.py

app.py

@@ -43,15 +43,27 @@ def blurr_object(image, object, blur_strength):
     blur_image = np.asarray(image.filter(ImageFilter.GaussianBlur(radius=blur_strength)))
     mask_array_three_channel_invert = 1-mask_array_three_channel
     blur_image_reverse_mask = blur_image*mask_array_three_channel_invert
-
-    seg_out=segmented_image.astype(np.uint8)
+    seg_box=[]
+    for i in range(object):
+        object_number = int(object.split('i')[0])
+        mask_array = np.asarray(pred[object_number]['mask'])/255
+        image_array = np.asarray(image)
+    
+        mask_array_three_channel = np.zeros_like(image_array)
+        mask_array_three_channel[:,:,0] = mask_array
+        mask_array_three_channel[:,:,1] = mask_array
+        mask_array_three_channel[:,:,2] = mask_array
+    
+        segmented_image = image_array*mask_array_three_channel
+        seg_out=segmented_image.astype(np.uint8)
+        seg_box.append(seg_out)
     #seg_out=image_array.astype(np.uint8)
 
     
     blurred_output_image = Image.fromarray((blur_image_reverse_mask).astype(np.uint8)+segmented_image.astype(np.uint8))
     for _ in range(int(blur_strength//2.5)):
         blurred_output_image = blurred_output_image.filter(ImageFilter.SMOOTH_MORE)
-    return blurred_output_image, seg_out
+    return blurred_output_image, seg_box
 
 app = gr.Blocks()