diff --git a/pyproject.toml b/pyproject.toml index b056cb7..61d488a 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -25,6 +25,10 @@ classifiers = [ "Topic :: Scientific/Engineering :: Image Processing", ] +[project.optional-dependencies] +OCR = ["torch <= 2.0.1", "transformers <= 4.30.2"] +plotting = ["matplotlib"] + [project.scripts] eynollah = "eynollah.cli:main" ocrd-eynollah-segment = "eynollah.ocrd_cli:main" diff --git a/requirements.txt b/requirements.txt index 02450aa..d72df29 100644 --- a/requirements.txt +++ b/requirements.txt @@ -4,8 +4,4 @@ numpy <1.24.0 scikit-learn >= 0.23.2 tensorflow < 2.13 imutils >= 0.5.3 -matplotlib -setuptools >= 50 -transformers <= 4.30.2 -torch <= 2.0.1 numba <= 0.58.1