Update app.py

app.py

@@ -24,35 +24,6 @@ def image_objects(image):
     pred = model(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)
-
-    object_number = int(object.split('_')[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
-
-    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=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
-
     
 def get_seg(image, object):
     image = img_resize(image)
@@ -94,30 +65,14 @@ with app:
             ### Input Image
             """)
             image_input = gr.Image(type="pil")
-            
-
+    with gr.Row():
         with gr.Column():
-            with gr.Row():
-                gr.Markdown(
-                """
-                ### Found Objects
-                """)
-            with gr.Row():
-                blur_slider = gr.Slider(minimum=0.5, maximum=10, value=3, label="Adject Blur Strength")
-            with gr.Row():
                 object_output = gr.Dropdown(label="Select Object From Dropdown")
-
     
     with gr.Row():
-        with gr.Column():
-            gr.Markdown(
-                """
-                ### Blurred Image Output
-                """)
-            image_output = gr.Image()
         with gr.Column():
             seg_btn = gr.Button(label="Run")
-            gal1=gr.Gallery(type="filepath").style(grid=10)
+            gal1=gr.Gallery(type="filepath").style(grid=4)
 
     seg_btn.click(get_seg, inputs=[image_input,object_output], outputs=gal1)
 
@@ -131,9 +86,4 @@ with app:
         inputs=[image_input, object_output, blur_slider],
         outputs=[image_output])
 
-    blur_slider.change(fn=blurr_object, 
-        inputs=[image_input, object_output, blur_slider],
-        outputs=[image_output])
-    
-
 app.launch()