version: "3.8"
services:
app:
build: .
command: python -m flask run --host=0.0.0.0
deploy:
resources:
reservations:
devices:
- driver: nvidia
count: all
capabilities: [ gpu ]
ports:
- "5000:5000"
volumes:
- .:/classla-api