diff --git a/docker/Dockerfile b/docker/Dockerfile
index b3e5d41..58ed5cb 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -1,11 +1,17 @@
+
+# For those who wanst to run on cpu only have to comment the following FROM and instead 
+# FROM debian:bullseye #(find the right OS description by $lsb_release -a)
 FROM nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04
 # replace CUDA version to your CUDA version.
 # You can check your CUDA version with below.
 # nvcc -V
 
+
 RUN apt-get update
 RUN apt-get install -y python3
 RUN apt-get -y install python3-pip git
+# python-multipart was missing which caused the docker to exit soon after start
+RUN pip install python-multipart
 RUN pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
 # replace CUDA version to your CUDA version.
 
@@ -16,6 +22,10 @@ RUN pip3 install fastapi uvicorn[standard] fsspec[http]==2023.1.0
 RUN git clone https://github.com/facebookresearch/nougat.git
 WORKDIR /workspace/nougat
 
+# The following upgrade for numpy and fsspec was required
+RUN pip3 install --no-cache-dir --upgrade numpy
+RUN pip3 install --no-cache-dir --upgrade "fsspec[http]>=2023.1.0,<=2023.10.0"
+
 RUN python3 setup.py install
 
 EXPOSE 8503