diff --git a/poetry.lock b/poetry.lock
index 8008429..963641f 100644
--- a/poetry.lock
+++ b/poetry.lock
@@ -4352,13 +4352,13 @@ files = [
 
 [[package]]
 name = "pyinfra"
-version = "2.0.0"
+version = "2.1.0"
 description = ""
 optional = false
 python-versions = ">=3.10,<3.11"
 files = [
-    {file = "pyinfra-2.0.0-py3-none-any.whl", hash = "sha256:cf50ba9e6d8491d41e3475bef103f4a773b67a86985b2c985516712679e19744"},
-    {file = "pyinfra-2.0.0.tar.gz", hash = "sha256:521d399d0d2900f6b8d82c5ce78ae41f662e9022a09bb22dc14e7b7d15e66d1b"},
+    {file = "pyinfra-2.1.0-py3-none-any.whl", hash = "sha256:6a51e60648a0c3ff8485d8da01cde03630ff861722499c5026dd6cfeb0fcf6f6"},
+    {file = "pyinfra-2.1.0.tar.gz", hash = "sha256:d4834c4a4a5b8a33974b05a50ac6b54d2f0e283af93da7787e24dacdb7b9d468"},
 ]
 
 [package.dependencies]
@@ -5874,4 +5874,4 @@ testing = ["big-O", "jaraco.functools", "jaraco.itertools", "more-itertools", "p
 [metadata]
 lock-version = "2.0"
 python-versions = ">=3.10,<3.11"
-content-hash = "af7acdc34d32780fd3da2000db7d24c7ba18104861bd931d2bf3f244b46dc8c6"
+content-hash = "10cfb6a6b2349ab0304084709b3fe6570b3854874d67cd0600eb831e8289377d"
diff --git a/pyproject.toml b/pyproject.toml
index 59fb551..da5323d 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "image-classification-service"
-version = "2.2.0"
+version = "2.4.0"
 description = ""
 authors = ["Team Research <research@knecon.com>"]
 readme = "README.md"
@@ -8,7 +8,7 @@ packages = [{ include = "image_prediction", from = "src" }]
 
 [tool.poetry.dependencies]
 python = ">=3.10,<3.11"
-pyinfra = { version = "2.0.0", source = "gitlab-research" }
+pyinfra = { version = "2.1.0", source = "gitlab-research" }
 kn-utils = { version = "0.2.7", source = "gitlab-research" }
 dvc = "^2.34.0"
 dvc-ssh = "^2.20.0"