diff --git a/Dockerfile b/Dockerfile
index e6bb7ec..7a4cf2e 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -32,4 +32,4 @@ WORKDIR /swarm_root/servers
 EXPOSE 8000
 
 # Command to run the application
-CMD ["python3.10", "-m", "uvicorn", "servers/cogvlm:cogvlm", "--host", "0.0.0.0", "--port", "8000"]
+CMD ["python3.10", "-m", "uvicorn", "servers/cogvlm:main", "--host", "0.0.0.0", "--port", "8000"]