FROM ghcr.io/immich-app/immich-machine-learning:release

ENV TRANSFORMERS_CACHE=/cache
ENV HF_HOME=/cache
ENV TORCH_HOME=/cache

EXPOSE 7896:3003

CMD ["python3", "-m", "immich_ml"]