Update main.py

main.py

@@ -133,7 +133,7 @@ async def segment_automatic(file: UploadFile = File(...)):
         
         # Return top masks
         result_masks = []
-        for i, mask_data in enumerate(masks[:10]):  # Return top 10 masks
+        for i, mask_data in enumerate(masks[:2]):  # Return top 10 masks
             mask = mask_data['segmentation']
             result_masks.append({
                 "id": i,