diff --git a/.github/workflows/gpu-integ-test.yaml b/.github/workflows/gpu-integ-test.yaml
index bf2eaf99..bcc3b986 100644
--- a/.github/workflows/gpu-integ-test.yaml
+++ b/.github/workflows/gpu-integ-test.yaml
@@ -93,7 +93,7 @@ jobs:
     name: Stop self-hosted EC2 runner
     needs:
       - start-runner
-      # - pytorch-integration-test
+      - pytorch-integration-test
       - tensorflow-integration-test
     runs-on: ubuntu-latest
     env:
diff --git a/dockerfiles/starlette/pytorch/Dockerfile.cpu b/dockerfiles/starlette/pytorch/Dockerfile.cpu
index 877ea3a6..03c0d9a6 100644
--- a/dockerfiles/starlette/pytorch/Dockerfile.cpu
+++ b/dockerfiles/starlette/pytorch/Dockerfile.cpu
@@ -1,4 +1,4 @@
-FROM huggingface/transformers-inference:4.20.1-pt1.11-cpu
+FROM huggingface/transformers-inference:4.23.1-pt1.11-cpu
 
 # install starlette framework
 COPY starlette_requirements.txt /tmp/requirements.txt
diff --git a/dockerfiles/starlette/pytorch/Dockerfile.gpu b/dockerfiles/starlette/pytorch/Dockerfile.gpu
index 6da7b455..69ab5de3 100644
--- a/dockerfiles/starlette/pytorch/Dockerfile.gpu
+++ b/dockerfiles/starlette/pytorch/Dockerfile.gpu
@@ -1,4 +1,4 @@
-FROM huggingface/transformers-inference:4.20.1-pt1.11-cuda11.5
+FROM huggingface/transformers-inference:4.23.1-pt1.11-cuda11.5
 
 # install starlette framework
 COPY starlette_requirements.txt /tmp/requirements.txt
diff --git a/dockerfiles/starlette/tensorflow/Dockerfile.cpu b/dockerfiles/starlette/tensorflow/Dockerfile.cpu
index d551c47d..b18d640b 100644
--- a/dockerfiles/starlette/tensorflow/Dockerfile.cpu
+++ b/dockerfiles/starlette/tensorflow/Dockerfile.cpu
@@ -1,4 +1,4 @@
-FROM huggingface/transformers-inference:4.21.1-tf2.9-cpu
+FROM huggingface/transformers-inference:4.23.1-tf2.9-cpu
 
 # install starlette framework
 COPY starlette_requirements.txt /tmp/requirements.txt
diff --git a/dockerfiles/starlette/tensorflow/Dockerfile.gpu b/dockerfiles/starlette/tensorflow/Dockerfile.gpu
index 5ace9ec8..82fc99e0 100644
--- a/dockerfiles/starlette/tensorflow/Dockerfile.gpu
+++ b/dockerfiles/starlette/tensorflow/Dockerfile.gpu
@@ -1,4 +1,4 @@
-FROM huggingface/transformers-inference:4.21.1-tf2.9-cuda11.2
+FROM huggingface/transformers-inference:4.23.1-tf2.9-cuda11.2
 
 # install starlette framework
 COPY starlette_requirements.txt /tmp/requirements.txt
diff --git a/tests/integ/config.py b/tests/integ/config.py
index 8a9b5610..d427fc1f 100644
--- a/tests/integ/config.py
+++ b/tests/integ/config.py
@@ -78,7 +78,7 @@
         "tensorflow": None,
     },
     "table-question-answering": {
-        "pytorch": "microsoft/tapex-base-finetuned-wtq",
+        "pytorch": "philschmid/tapex-tiny",
         "tensorflow": None,
     },
     "zero-shot-image-classification": {