diff --git a/Dockerfile b/Dockerfile
index 1423c29..506397e 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -20,7 +20,7 @@ WORKDIR /app/TensorRT-LLM/examples/llm-api
 RUN pip3 install -r requirements.txt
 
 # Install additional dependencies for the serverless worker
-RUN pip3 install runpod
+RUN pip3 install --upgrade runpod transformers
 
 # Set the working directory to /app
 WORKDIR /app