Update app.py

app.py

@@ -39,13 +39,17 @@ def create_segmentation(mask_np_list):
 
 
 def run_segmentation_wrapper(image):
-    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
-    sliderup = gr.Slider(value = 0, minimum=0, maximum=max_val, step=1, visible=True)
-    return image, segmentation, mask_np_list, mask_label_list, image, sliderup, sliderup , 'Segmentation finish. Select mask id and move to the next step.'
+    try:
+        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
+        sliderup = gr.Slider(value = 0, minimum=0, maximum=max_val, step=1, visible=True)
+        return image, segmentation, mask_np_list, mask_label_list, image, sliderup, sliderup , 'Segmentation finish. Select mask id and move to the next step.'
+    except:
+        sliderup = gr.Slider(value = 0, minimum=0, maximum=max_val, step=1, visible=False)
+        return None,None,None,None,None, sliderup, sliderup , 'Please upload an image before proceeding.'
         
 
 def transparent_paste_with_mask(backimg, foreimg, mask_np,transparency = 128):