Image build is optimized.

flask_ml_api/Dockerfile

@@ -16,7 +16,7 @@ COPY . /app
 
 # Install dependencies in separate steps to reduce memory usage
 RUN apk add --no-cache gcompat
-RUN /opt/myenv/bin/pip install torch && \
+RUN /opt/myenv/bin/pip install torch --index-url https://download.pytorch.org/whl/cpu && \
     rm -rf /root/.cache/pip \
 RUN /opt/myenv/bin/pip install Flask && \
     rm -rf /root/.cache/pip