Peng Shiya commited on
Commit
cba1a87
·
1 Parent(s): 50a5f00

fix: improved error logging

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -84,8 +84,7 @@ with block:
84
  if configs.enable_segment_all:
85
  masks, _ = service.predict_all(sam, image)
86
  else:
87
- gr.Warning('Segment-all disabled, set point label(s) before running')
88
- return inputs[cutout_gallery], inputs[masks_annotated_image]
89
  else:
90
  masks, _ = service.predict_conditioned(sam,
91
  image,
 
84
  if configs.enable_segment_all:
85
  masks, _ = service.predict_all(sam, image)
86
  else:
87
+ raise gr.Error('Segment-all disabled, set point label(s) before running')
 
88
  else:
89
  masks, _ = service.predict_conditioned(sam,
90
  image,