Fix path

prismer_model.py

@@ -7,12 +7,12 @@ import sys
 import cv2
 import torch
 
-from prismer.dataset import create_dataset, create_loader
-from prismer.model.prismer_caption import PrismerCaption
-
-
 repo_dir = pathlib.Path(__file__).parent
 submodule_dir = repo_dir / 'prismer'
+sys.path.insert(0, submodule_dir.as_posix())
+
+from dataset import create_dataset, create_loader
+from model.prismer_caption import PrismerCaption
 
 
 def download_models() -> None:
@@ -50,11 +50,11 @@ def run_experts(image_path: str) -> tuple[str | None, ...]:
     for expert_name in expert_names:
         env = os.environ.copy()
         if 'PYTHONPATH' in env:
-            env['PYTHONPATH'] = f'{repo_dir.as_posix()}:{env["PYTHONPATH"]}'
+            env['PYTHONPATH'] = f'{submodule_dir.as_posix()}:{env["PYTHONPATH"]}'
         else:
-            env['PYTHONPATH'] = repo_dir.as_posix()
+            env['PYTHONPATH'] = submodule_dir.as_posix()
 
-        subprocess.run(shlex.split(f'python prismer/experts/generate_{expert_name}.py'), env=env, check=True)
+        subprocess.run(shlex.split(f'python experts/generate_{expert_name}.py'), cwd='prismer', env=env, check=True)
 
     # keys = ['depth', 'edge', 'normal', 'seg_coco', 'obj_detection', 'ocr_detection']
     keys = ['depth', 'edge', 'normal']