Fix parallization

prismer_model.py

@@ -63,10 +63,12 @@ def run_experts(image_path: str) -> tuple[str | None, ...]:
     out_path = image_dir / 'image.jpg'
     cv2.imwrite(out_path.as_posix(), cv2.imread(image_path))
 
-    expert_names = ['edge', 'normal', 'objdet', 'ocrdet', 'segmentation']
     run_expert('depth')
     with concurrent.futures.ProcessPoolExecutor() as executor:
-        executor.map(run_expert, expert_names)
+        executor.map(run_expert, ['edge', 'normal', 'objdet'])
+
+    with concurrent.futures.ProcessPoolExecutor() as executor:
+        executor.map(run_expert, ['ocrdet', 'segmentation'])
 
     keys = ['depth', 'edge', 'normal', 'seg_coco', 'obj_detection', 'ocr_detection']
     results = [pathlib.Path('prismer/helpers/labels') / key / 'helpers/images/image.png' for key in keys]