niulx commited on
Commit
59ced18
1 Parent(s): cda7518

Update segment.py

Browse files

segmentation show process

Files changed (1) hide show
  1. segment.py +1 -1
segment.py CHANGED
@@ -117,4 +117,4 @@ def run_segmentation(image, name="example_tmp", size = 512, noseg=False):
117
  draw_panoptic_segmentation(**panoptic_segmentation, save_folder = save_folder, noseg = noseg, model = model)
118
  print("Finish segment")
119
  #block_flag += 1
120
- return gr.Button("1.2 Load original masks",visible = True)#, gr.Button.update("1.2 Load edited masks",visible = True), gr.Checkbox.update(label = "Show Segmentation",visible = True)
 
117
  draw_panoptic_segmentation(**panoptic_segmentation, save_folder = save_folder, noseg = noseg, model = model)
118
  print("Finish segment")
119
  #block_flag += 1
120
+ return gr.Button("1.2 Load original masks",visible = True), "Segmentation finished."#, gr.Button.update("1.2 Load edited masks",visible = True), gr.Checkbox.update(label = "Show Segmentation",visible = True)