diff --git a/docker-compose.yml b/docker-compose.yml
index 2363a35..645a613 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -12,6 +12,9 @@
 
 services:
   aiflow:
+    build:
+      context: .
+      dockerfile: Dockerfile
     image: qihuaiflow:with-deps
     container_name: qihu-aiflow
     ports: