version: "3.7"

services:
  model-server:
    image: your-docker-image-name
    environment:
      BASIC_AUTH_USER: "konze"
      BASIC_AUTH_PASS: "konze@123"
      LOG_LEVEL: "info"
      WORKERS: 4
      THREADS: 2
      BASELINE_MODEL_NAME: "PravinKonze321/course_ner"
      FINETUNED_MODEL_NAME: "finetuned_model"
      LABEL_STUDIO_HOST: "http://192.168.2.88:8080"
      LABEL_STUDIO_API_KEY: "d42beb6660a41e1baa13ad0b10c3547efbc5f264"
      START_TRAINING_EACH_N_UPDATES: 10
      LEARNING_RATE: 0.001
      NUM_TRAIN_EPOCHS: 10
      WEIGHT_DECAY: 0.01
      MODEL_DIR: "./results"
    volumes:
      - ./results:/app/results
    ports:
      - "9090:9090"