diff --git a/.github/workflows/main.yaml b/.github/workflows/main.yaml
index 1e75b26..a66c493 100644
--- a/.github/workflows/main.yaml
+++ b/.github/workflows/main.yaml
@@ -35,4 +35,4 @@ jobs:
         with:
           context: .
           push: true
-          tags: hal9ai/hal9-docker:latest,hal9ai/hal9-docker:0.2.23
+          tags: hal9ai/hal9-docker:latest,hal9ai/hal9-docker:0.2.24
diff --git a/requirements.txt b/requirements.txt
index c0ecf5c..f73ea57 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -68,4 +68,5 @@ fastapi
 aiofiles
 redis
 transformers
-typing
\ No newline at end of file
+typing
+replicate==1.0.3
\ No newline at end of file