Update app.py

app.py

@@ -119,8 +119,8 @@ mask_adapter = None
 @torch.no_grad()
 @torch.autocast(device_type="cuda", dtype=torch.float32)
 def inference_box(input_img,  img_state,):
-
-
+    if len(img_state.selected_bboxes) != 2:
+        return None
     mp.set_start_method("spawn", force=True)
     
     box_points = img_state.selected_bboxes
@@ -254,10 +254,9 @@ examples = [
 
 examples_point = [
     ['./demo/images/ADE_val_00000739.jpg'],
-    ['./demo/images/000000052462.jpg'],
-    ['./demo/images/000000081766.jpg'],
+    ['./demo/images/000000290833.jpg'],
+    ['./demo/images/2010_001315.jpg'],
     ['./demo/images/ADE_val_00000001.jpg'],
-    ['./demo/images/000000033707.jpg'],
     ['./demo/images/000000000785.jpg'],
 ]