bug fix.

app.py

@@ -64,11 +64,8 @@ def load_image_ui(load_edit, input_folder="example_tmp"):
 
 
 def run_segmentation_wrapper(image):
-    mask_np_list,mask_label_list = run_segmentation(image)
-    for img_path in Path("example_tmp").iterdir():
-        if img_path.name in ["img_512.png"]:
-            image = Image.open(img_path)
-    image = image.convert('RGB')
+    image, mask_np_list,mask_label_list = run_segmentation(image)
+    #image = image.convert('RGB')
     segmentation = create_segmentation(mask_np_list)
     print("!!", len(mask_np_list))  
     max_val = len(mask_np_list)-1