diff --git a/imageserver/Dockerfile b/imageserver/Dockerfile
index ddc184b..060ca02 100644
--- a/imageserver/Dockerfile
+++ b/imageserver/Dockerfile
@@ -1,8 +1,14 @@
 FROM nvidia/cuda:11.6.0-base-ubuntu20.04
 RUN apt-get update && apt-get install python3 python3-pip -y
 
-RUN pip3 install --upgrade diffusers transformers scipy python-dotenv cuda-python && \
+RUN pip3 install --upgrade diffusers transformers scipy python-dotenv cuda-python fastapi uvicorn && \
     pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117 && \
     pip install torch==1.11.0+cu115 torchvision==0.12.0+cu115 torchaudio==0.11.0+cu115 -f https://download.pytorch.org/whl/torch_stable.html
 
-COPY main.py /main.py
\ No newline at end of file
+WORKDIR /app
+
+COPY main.py /app/main.py
+VOLUME /root/.cache/huggingface/diffusers/
+
+
+CMD [ "uvicorn", "main:app", "--host", "0.0.0.0" ]
\ No newline at end of file