diff --git a/docker/docker-compose.yml b/docker/docker-compose.yml
index 50b1401..6ee578d 100644
--- a/docker/docker-compose.yml
+++ b/docker/docker-compose.yml
@@ -10,7 +10,7 @@ services:
     working_dir: /app
     ports:
       - "9000:8000"
-    command: bash -c "python api.py"
+    command: bash -c "python web_demo.py"
     deploy:
       resources:
         reservations:
diff --git a/docker/dockerfile b/docker/dockerfile
index 79d5acd..6b49c15 100644
--- a/docker/dockerfile
+++ b/docker/dockerfile
@@ -9,8 +9,10 @@ RUN apt update && apt-get -y install git wget \
 RUN ln -s /usr/bin/python3.10 /usr/bin/python
 RUN useradd -ms /bin/bash usera
 WORKDIR /app
-ADD . .
+ADD .requirements .
 RUN pip config set global.index-url https://mirrors.cloud.tencent.com/pypi/simple
 RUN pip config set global.trusted-host mirrors.cloud.tencent.com
 RUN pip install -r requirements.txt --no-cache-dir
-
+ADD . .
+# preinstall model
+RUN python -c "from transformers import AutoTokenizer, AutoModel; AutoModel.from_pretrained('THUDM/chatglm2-6b', trust_remote_code=True)"
\ No newline at end of file