version: '3.8'

services:
  bert-api:
    build: .
    ports:
      - "7860:7860"
    volumes:
      - ../saved_models:/app/saved_models
      - ../tokenizer:/app/tokenizer
      - ../predictions:/app/predictions
      - ../label_encoders.pkl:/app/label_encoders.pkl
      - ../.cache:/app/.cache
    environment:
      - PYTHONUNBUFFERED=1
      - TRANSFORMERS_CACHE=/app/.cache
      - PORT=7860
    restart: unless-stopped