docker-compose.yml

version: '3.8'

services:
  huggingface-android:
    build: .
    image: huggingface-android:latest
    container_name: huggingface-android
    environment:
      - HUGGINGFACE_TOKEN=${HUGGINGFACE_TOKEN}
    volumes:
      - ./models:/workspace/models
      - ./android-project:/workspace/android-project
      - ~/.gradle:/root/.gradle
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: all
              capabilities: [gpu]
    tty: true
    stdin_open: true