FROM rust:latest as builder
WORKDIR /usr/src/app
RUN apt update \
	&& apt install -y --no-install-recommends python3-dev \
	&& rm -rf /var/lib/apt/lists/*
COPY ./dermy-api/ .
COPY ./dermy-models/models/mobilenet_v3.keras ./src/model/model.keras
RUN cargo build --release

FROM tensorflow/tensorflow:2.16.1
COPY ./dermy-models/requirements.txt .
RUN pip install --no-cache-dir -r requirements.txt 
COPY --from=builder /usr/src/app/target/release/dermy-api /usr/local/bin/dermy-api

ENTRYPOINT dermy-api