kevin.yang 6 months ago
parent
commit
26e4887edb
2 changed files with 7 additions and 21 deletions
  1. 1 3
      Dockerfile
  2. 6 18
      docker-compose.yml

+ 1 - 3
Dockerfile

@@ -11,6 +11,4 @@ RUN apt update && apt --no-install-recommends install -y libgl-dev libglib2.0-de
     && pip install -r requirements.txt --no-cache-dir \
     && mkdir -p /app/cache
 
-EXPOSE 8080
-
-ENTRYPOINT ["python", "/app/analyze_video.py"]
+ENTRYPOINT ["python", "/app/job.py"]

+ 6 - 18
docker-compose.yml

@@ -1,41 +1,29 @@
-version: '3.9'
-
 services:
-  server:
+  worker1:
     build:
       context: .
       dockerfile: Dockerfile
     image: google_ai_studio
-    container_name: google_server
-    restart: unless-stopped
-    environment:
-      - env=prod
-    networks:
-      - google_net
-  worker1:
-    depends_on:
-      - server
-    image: google_ai_studio
     container_name: google_worker1
     restart: unless-stopped
     environment:
-      - env=prod
+      - ENV=prod
       - API_KEY=AIzaSyB2kjF2-S2B5cJiosx_LpApd227w33CVvs
+      - TASK_TYPE=recommend
     networks:
       - google_net
-    entrypoint: "python /app/job_video_processing.py"
   worker2:
     depends_on:
-      - server
+      - worker1
     image: google_ai_studio
     container_name: google_worker2
     restart: unless-stopped
     environment:
-      - env=prod
+      - ENV=prod
       - API_KEY=AIzaSyCor0q5w37Dy6fGxloLlCT7KqyEFU3PWP8
+      - TASK_TYPE=top
     networks:
       - google_net
-    entrypoint: "python /app/job_video_processing.py"
 
 networks:
   google_net: