Update app.py

app.py

@@ -24,6 +24,8 @@ def image_objects(image):
     pred_object_list = [str(i)+'_'+x['label'] for i, x in enumerate(pred)]
     return gr.Dropdown.update(choices = pred_object_list, interactive = True)
 
+
+
 def blurr_object(image, object, blur_strength):
     image = img_resize(image)
 
@@ -45,7 +47,7 @@ def blurr_object(image, object, blur_strength):
     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
+    return blurred_output_image, segmented_image
 
 app = gr.Blocks()
 
@@ -85,7 +87,8 @@ with app:
                 ### Blurred Image Output
                 """)
             image_output = gr.Image()
-        
+        with gr.Column():
+            gal1=gr.Gallery()
 
     image_input.change(fn=image_objects, 
         inputs=image_input, 
@@ -94,7 +97,7 @@ with app:
 
     object_output.change(fn=blurr_object, 
         inputs=[image_input, object_output, blur_slider],
-        outputs=image_output)
+        outputs=image_output, gal1)
 
     blur_slider.change(fn=blurr_object, 
         inputs=[image_input, object_output, blur_slider],