63 lines
1.4 KiB
YAML
63 lines
1.4 KiB
YAML
version: '3.8'
|
|
|
|
networks:
|
|
emom_mesh:
|
|
driver: bridge
|
|
|
|
services:
|
|
emom_ui:
|
|
build:
|
|
context: .
|
|
dockerfile: docker/Dockerfile.ui
|
|
container_name: emom_web_ui
|
|
restart: unless-stopped
|
|
ports:
|
|
- "8080:8080"
|
|
networks:
|
|
- emom_mesh
|
|
env_file:
|
|
- .env
|
|
volumes:
|
|
- ./src:/app/src
|
|
- ${DATA_DEAM_DIR}:/app/dataset/DEAM:ro
|
|
depends_on:
|
|
- emom_inference
|
|
|
|
emom_inference:
|
|
build:
|
|
context: .
|
|
dockerfile: docker/Dockerfile.api
|
|
container_name: emom_pytorch_api
|
|
restart: unless-stopped
|
|
networks:
|
|
- emom_mesh
|
|
env_file:
|
|
- .env
|
|
volumes:
|
|
- ${HOST_ARTIFACTS_DIR}/emoset_resnet50_best.pth:/app/src/emoset_resnet50_best.pth:ro
|
|
- ${HOST_ARTIFACTS_DIR}/music_engine/va_regressor.pkl:/app/src/music_engine/va_regressor.pkl:ro
|
|
- ${DATA_DEAM_DIR}:/app/dataset/DEAM:ro
|
|
- ~/.cache/huggingface:/root/.cache/huggingface
|
|
deploy:
|
|
resources:
|
|
reservations:
|
|
devices:
|
|
- driver: nvidia
|
|
count: 1
|
|
capabilities: [gpu]
|
|
|
|
emom_ollama:
|
|
image: ollama/ollama:latest
|
|
container_name: emom_ollama_engine
|
|
restart: unless-stopped
|
|
networks:
|
|
- emom_mesh
|
|
volumes:
|
|
- ~/.ollama:/root/.ollama
|
|
deploy:
|
|
resources:
|
|
reservations:
|
|
devices:
|
|
- driver: nvidia
|
|
count: 1
|
|
capabilities: [gpu] |