Upload script.py

script.py

@@ -6,7 +6,6 @@ import tqdm
 
 BASE = Path(__file__).resolve().parents[0]
 DATASET_DIR = BASE / "dataset"
-GPT_DIR = BASE / "gpt"
 
 
 def find_split_dir() -> Path:
@@ -22,10 +21,10 @@ def run_for_pdf(pdf_path: Path, out_path: Path) -> int:
     out_path.parent.mkdir(parents=True, exist_ok=True)
 
     cmd = [sys.executable, "main.py", "--pdf", str(pdf_path), "--out", str(out_path)]
-    print(f"Running: {' '.join(cmd)} (cwd={GPT_DIR})")
-    proc = subprocess.run(cmd, cwd=str(GPT_DIR), stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
+    print(f"Running: {' '.join(cmd)}")
+    proc = subprocess.run(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
     if proc.returncode != 0:
-        print(f"ERROR: gpt/main.py failed for {pdf_path.name} (rc={proc.returncode})")
+        print(f"ERROR: main.py failed for {pdf_path.name} (rc={proc.returncode})")
         print(proc.stdout)
         print(proc.stderr)
     else: