diff --git a/Dockerfile b/Dockerfile
index 262f112..1caa9c5 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,5 +1,5 @@
 # Stage 1: Build stage
-FROM bitnami/pytorch AS builder
+FROM python:slim AS builder
 
 WORKDIR /app