| 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 |