Update classify.py

classify.py

@@ -8,7 +8,7 @@ import statistics
 import pandas as pd
 import multiprocessing as mp
 from functools import lru_cache
-from concurrent.futures import ThreadPoolExecutor
+from concurrent.futures import ThreadPoolExecutor, ProcessPoolExecutor
 from processor_regex import classify_with_regex
 from processor_bert  import classify_batch as bert_batch
 from processor_llm   import classify_with_llm
@@ -118,10 +118,10 @@ def classify_csv(input_path: str, output_path: str = "output.csv") -> tuple[str,
     
     t_start = time.perf_counter()
     
-    # FIX: Use 'spawn' context! This is the magic that prevents PyTorch/ONNX Segfaults
+    # FIX: Correctly pass the spawn context to ProcessPoolExecutor
     ctx = mp.get_context('spawn')
     
-    with ctx.ProcessPoolExecutor(max_workers=safe_cores) as executor:
+    with ProcessPoolExecutor(max_workers=safe_cores, mp_context=ctx) as executor:
         for chunk_result in executor.map(_process_chunk, chunks):
             results.extend(chunk_result)