Update app.py

app.py

@@ -39,6 +39,7 @@ def create_segmentation(mask_np_list):
 #@spaces.GPU
 def run_segmentation_wrapper(image):
     try:
+        print(image.shape)
         image, mask_np_list,mask_label_list = run_segmentation(image)
         #image = image.convert('RGB')
         segmentation = create_segmentation(mask_np_list)
@@ -47,7 +48,8 @@ def run_segmentation_wrapper(image):
         sliderup = gr.Slider(value = 0, minimum=0, maximum=max_val, step=1, visible=True)
         gr.Info('Segmentation finish. Select mask id and move to the next step.')
         return image, segmentation, mask_np_list, mask_label_list, image, sliderup, sliderup , 'Segmentation finish. Select mask id and move to the next step.'
-    except:
+    except Exception as e:
+        print(e)
         sliderup = gr.Slider(value = 0, minimum=0, maximum=1, step=1, visible=False)
         gr.Warning('Please upload an image before proceeding.')
         return None,None,None,None,None, sliderup, sliderup , 'Please upload an image before proceeding.'