Update app.py

app.py

@@ -27,8 +27,8 @@ def image_objects(image):
     
 def get_seg(image, object):
     image = img_resize(image)
-
-    #blur_image_reverse_mask = blur_image*mask_array_three_channel_invert
+    pred = model(image)
+    pred_object_list = [str(i)+'_'+x['label'] for i, x in enumerate(pred)]
     seg_box=[]
     for i in range(len(pred)):
         #object_number = int(object.split('_')[0])
@@ -44,7 +44,8 @@ def get_seg(image, object):
         seg_out=segmented_image.astype(np.uint8)
         seg_box.append(seg_out)  
 
-    return(seg_box)
+    return(seg_box,gr.Dropdown.update(choices = pred_object_list, interactive = True))
+)
 
 
 
@@ -54,9 +55,8 @@ app = gr.Blocks()
 with app:
     gr.Markdown(
             """
-            ## Portrait Photo Generator
-            - Create stunning portrait photos by blurring the background of your selected object.
-            - Adjust the blurring strength using the slider.
+            ## Image Dissector 
+            
             """)
     with gr.Row():
         with gr.Column():
@@ -75,12 +75,9 @@ with app:
             gal1=gr.Gallery(type="filepath").style(grid=4)
 
     seg_btn.click(get_seg, inputs=[image_input,object_output], outputs=gal1)
-
+    image_input.change(get_seg, inputs=[image_input], outputs=[gal1,object_output])
     
-    image_input.change(fn=image_objects, 
-        inputs=image_input, 
-        outputs=object_output
-        )
+    #image_input.change(fn=image_objects,inputs=image_input,outputs=object_output)