diff --git a/Dockerfile b/Dockerfile
index 9131487..80abc14 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -5,4 +5,4 @@ ENV PATH=/torch/venv3/pytorch_infer/bin:/workspace/ffmpeg-mlu-v4.2.0/install/bin
 
 RUN pip install diffusers==0.34.0
 RUN sed -i 's|source /torch/venv3/pytorch/bin/activate|source /torch/venv3/pytorch_infer/bin/activate|' /root/.bashrc
-COPY main.py test.sh dataset.json /workspace/
+COPY main.py dataset.json /workspace/