Fix path

gradio_caption.py

@@ -24,7 +24,7 @@ def create_demo():
                 object_detection = gr.Image(label='Object Detection')
                 ocr = gr.Image(label='OCR Detection')
 
-    inputs = [image, model_name]
+    inputs = [image]
     outputs = [depth, edge, normals]
 
     paths = sorted(pathlib.Path('prismer/images').glob('*'))

prismer_model.py

@@ -44,7 +44,8 @@ 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 = ['depth', 'edge', 'normal', 'objdet', 'ocrdet', 'segmentation']
+    # expert_names = ['depth', 'edge', 'normal', 'objdet', 'ocrdet', 'segmentation']
+    expert_names = ['depth', 'edge', 'normal']
     for expert_name in expert_names:
         env = os.environ.copy()
         if 'PYTHONPATH' in env: