diff --git a/api/Containerfile.cuda.ubuntu b/api/Containerfile.cuda.ubuntu
index 04713332..9e9bf7a0 100644
--- a/api/Containerfile.cuda.ubuntu
+++ b/api/Containerfile.cuda.ubuntu
@@ -1,11 +1,15 @@
 FROM docker.io/nvidia/cuda:11.7.1-runtime-ubuntu20.04
 
+WORKDIR /onnx-web/api
+
+RUN apt-get -y update \
+ && apt-get install -y ffmpeg libsm6 libxext6 \
+ && rm -rf /var/lib/apt/lists/*
+
 RUN apt-get -y update \
  && apt-get -y install python3 python3-pip python3-venv \
  && rm -rf /var/lib/apt/lists/*
 
-WORKDIR /onnx-web/api
-
 RUN python3 -m venv onnx_env
 
 ENV PATH="/onnx-web/api/onnx_web/bin:$PATH"
diff --git a/api/Containerfile.rocm.ubuntu b/api/Containerfile.rocm.ubuntu
index 9adf9e8b..a62995db 100644
--- a/api/Containerfile.rocm.ubuntu
+++ b/api/Containerfile.rocm.ubuntu
@@ -2,6 +2,10 @@ FROM docker.io/rocm/pytorch:rocm5.4_ubuntu20.04_py3.8_pytorch_1.12.1
 
 WORKDIR /onnx-web/api
 
+RUN apt-get -y update \
+ && apt-get install -y ffmpeg libsm6 libxext6 \
+ && rm -rf /var/lib/apt/lists/*
+
 RUN pip3 install virtualenv --upgrade
 
 RUN python3 -m venv onnx_env