diff --git a/Dockerfile b/Dockerfile
index f8cd9ac..eaa1ed9 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,6 +1,6 @@
 FROM pytorch/pytorch:1.12.1-cuda11.3-cudnn8-runtime
 WORKDIR /app
-COPY *.py requirements.txt /app
+COPY *.py requirements.txt *.pickle /app
 RUN apt update && apt install -y git
 RUN pip install -r requirements.txt
 ENV GRADIO_SERVER_PORT=7860