diff --git a/.github/workflows/docker-build-ec2.yml b/.github/workflows/docker-build-ec2.yml
index 0b1174fa..f0a1a555 100644
--- a/.github/workflows/docker-build-ec2.yml
+++ b/.github/workflows/docker-build-ec2.yml
@@ -41,6 +41,6 @@ jobs:
             sudo nvidia-docker build -t cogvlm . || exit 1
 
             # Run the image
-            sudo docker run -d -p 6092:6092 --gpus all --name cogvlm-container \
+            sudo docker run -d -p 7861:7861--gpus all --name cogvlm-container \
               -e OPENAI_API_KEY=${{ secrets.OPENAI_API_KEY }} \
-              cogvlm
\ No newline at end of file
+              cogvlm
diff --git a/Dockerfile b/Dockerfile
index d228a674..ad530b3b 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -14,10 +14,10 @@ RUN python -m pip install --upgrade pip
 RUN pip install --no-cache-dir -r requirements.txt
 
 # Make port 80 available to the world outside this container
-EXPOSE 80
+EXPOSE 7861
 
 # Define environment variable
 ENV NAME World
 
 # Run composite_demo/main.py when the container launches
-CMD ["python", "composite_demo/main.py"]
+CMD ["streamlit", "run", "composite_demo/main.py", "--server.port", "7861"]