docker-compose.yml

version: '3.8'

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