box first

app.py

@@ -72,6 +72,9 @@ def format_prompt_points(points):
             prompt_points.append([point[0], point[1]])
             label = 1 if point[2] == 1.0 else 0
             point_labels.append(label)
+    prompt_points = [[prompt_points]] if len(prompt_points) > 0 else None
+    point_labels = [point_labels] if len(point_labels) > 0 else None
+    prompt_boxes = [prompt_boxes] if len(prompt_boxes) > 0 else None
     return prompt_points, point_labels, prompt_boxes
 
 def segment_with_points(
@@ -85,9 +88,9 @@ def segment_with_points(
     print(prompt_points, point_labels, prompt_boxes)
     # segment
     inputs = processor(image, 
-                       input_boxes = [prompt_boxes],
-                       input_points=[[prompt_points]], 
-                       input_labels=[point_labels], 
+                       input_boxes = prompt_boxes,
+                       input_points=prompt_points, 
+                       input_labels=point_labels, 
                        return_tensors="pt").to(device)
     with torch.no_grad():
       outputs = model(**inputs)
@@ -140,4 +143,4 @@ with gr.Blocks(css=css, title='Segment Food with Prompts') as demo:
     clear_btn.click(clear, outputs=[candidate_pic, segpic_output1, segpic_output2, segpic_output3])
 
 demo.queue()
-demo.launch(share=True)
+demo.launch()