diff --git a/Dockerfile b/Dockerfile
index 05654db..6c3b5e5 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,12 +1,15 @@
 FROM rust:latest as builder
 WORKDIR /usr/src/app
-COPY ./dermy-api/* .
+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 python:alpine
+FROM tensorflow/tensorflow:2.16.1
 COPY ./dermy-models/requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt 
-COPY --from=builder ./target/release/dermy-api /usr/local/bin/dermy-api
+COPY --from=builder /usr/src/app/target/release/dermy-api /usr/local/bin/dermy-api
 
 ENTRYPOINT dermy-api
diff --git a/dermy-models b/dermy-models
index 2dd9768..0d8d48b 160000
--- a/dermy-models
+++ b/dermy-models
@@ -1 +1 @@
-Subproject commit 2dd97682d89968a98901d8c4b7c3d595825ee6c7
+Subproject commit 0d8d48b1aa7035e185afc7d573ccb18435e03b4e