version: '3'
services:
  pytorch:    
    image: "harbor.5jstore.com:8020/ai/wm_video_cut:v1"
    restart: always
    runtime: nvidia
    environment:
      - TZ=Asia/Shanghai
    volumes:
      - ./inputs/:/app/inputs/
      - ./outputs/:/app/outputs/
      - ./app/ai_gen_video/checkpoints/:/app/app/ai_gen_video/checkpoints/
      - ./app/ai_gen_image/checkpoints/:/app/app/ai_gen_image/checkpoints/
      #- ./gunicorn.conf.py:/app/gunicorn.conf.py
      #- ./start.py:/app/start.py
      #- ./app/:/app/app/
    ports:
      - "8383:5000"