Update app.py

app.py

@@ -44,7 +44,9 @@ def blurr_object(image, object, 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_out=segmented_image.astype(np.uint8)
+    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)):
@@ -90,7 +92,7 @@ with app:
                 """)
             image_output = gr.Image()
         with gr.Column():
-            gal1=gr.Pil()
+            gal1=gr.Gallery()
 
     image_input.change(fn=image_objects, 
         inputs=image_input,