File size: 2,526 Bytes
198ccb0 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 |
services:
# API Service
api:
build:
context: .
dockerfile: Dockerfile.api
container_name: news-classifier-api
ports:
- "8000:8000"
volumes:
# Mount only the required artifacts (avoid overriding code inside the image)
- ./models/distilmbert_lora_10k_v1.pt:/app/models/distilmbert_lora_10k_v1.pt:ro
- ./config/thresholds.json:/app/config/thresholds.json:ro
# Persist monitoring logs across restarts (optional but nice for demos)
- ./monitoring:/app/monitoring
environment:
- PYTHONUNBUFFERED=1
- MODEL_PATH=models/distilmbert_lora_10k_v1.pt
- THRESHOLDS_PATH=config/thresholds.json
- TOKENIZER_NAME=distilbert-base-multilingual-cased
restart: unless-stopped
healthcheck:
test: ["CMD", "curl", "-f", "http://localhost:8000/health"]
interval: 30s
timeout: 10s
retries: 3
start_period: 40s
networks:
- nlp-network
# Streamlit (multipage app)
streamlit:
build:
context: .
dockerfile: Dockerfile.streamlit
container_name: news-classifier-streamlit
ports:
- "8501:8501"
volumes:
- .:/app
working_dir: /app
environment:
- API_URL=http://api:8000
command: >
streamlit run streamlit_app.py
--server.port=8501
--server.address=0.0.0.0
depends_on:
- api
networks:
- nlp-network
# Optional: MLflow tracking server
mlflow:
image: python:3.10-slim
container_name: mlflow-server
ports:
- "5000:5000"
volumes:
- ./mlruns:/mlruns
- ./requirements.txt:/requirements.txt
working_dir: /app
command: >
sh -c "pip install mlflow &&
mlflow server --host 0.0.0.0 --port 5000 --backend-store-uri file:///mlruns"
environment:
- MLFLOW_BACKEND_STORE_URI=file:///mlruns
networks:
- nlp-network
profiles:
- tracking
# Optional: Jupyter notebook for development
jupyter:
build:
context: .
dockerfile: Dockerfile.dev
container_name: jupyter-notebook
ports:
- "8888:8888"
volumes:
- .:/app
- jupyter-data:/home/jovyan/work
environment:
- JUPYTER_ENABLE_LAB=yes
command: >
sh -c "pip install jupyter jupyterlab &&
jupyter lab --ip=0.0.0.0 --port=8888 --no-browser --allow-root --NotebookApp.token=''"
networks:
- nlp-network
profiles:
- dev
networks:
nlp-network:
driver: bridge
volumes:
jupyter-data:
|