diff --git a/ocr-service/main.py b/ocr-service/main.py index 955f7fbd..8e4d2288 100644 --- a/ocr-service/main.py +++ b/ocr-service/main.py @@ -366,7 +366,8 @@ async def train_model( os.makedirs(checkpoint_dir, exist_ok=True) cmd = [ - "ketos", "train", + "ketos", "--workers", "0", + "train", "-f", "path", "-o", checkpoint_dir, "-q", "fixed", @@ -456,7 +457,8 @@ async def segtrain_model( os.makedirs(checkpoint_dir, exist_ok=True) cmd = [ - "ketos", "segtrain", + "ketos", "--workers", "0", + "segtrain", "-o", checkpoint_dir, "-q", "fixed", "-N", "50",