Update app.py

app.py

@@ -49,7 +49,8 @@ def load_mask_ui(input_folder="example_tmp",load_edit = False):
     return mask_np_list, mask_label_list
 
 def load_image_ui(load_edit, input_folder="example_tmp"):
-    try:
+    #try:
+    if 1:
         for img_path in Path(input_folder).iterdir():
             if img_path.name in ["img_512.png"]:
                 image = Image.open(img_path)
@@ -60,9 +61,9 @@ def load_image_ui(load_edit, input_folder="example_tmp"):
         max_val = len(mask_np_list)-1
         sliderup = gr.Slider(value = 0, minimum=0, maximum=max_val, step=1, interactive=True)
         return image, segmentation, mask_np_list, mask_label_list, image, sliderup, sliderup 
-    except:
-        print("Image folder invalid: The folder should contain image.png")
-        return None, None, None, None, None, None
+    #except:
+    #    print("Image folder invalid: The folder should contain image.png")
+    #    return None, None, None, None, None, None