Update Dockerfile

Dockerfile

@@ -1,14 +1,19 @@
 # Dockerfile (Place in the root of your repo)
-FROM python:3.10-slim # Or a version compatible with your dependencies
+
+# Use a specific Python version slim image for smaller size
+# Or a version compatible with your dependencies
+FROM python:3.10-slim
 
 WORKDIR /code
 
 # Install system dependencies (like ffmpeg for imageio's mp4 writing)
+# Using --no-install-recommends helps keep the image smaller
 RUN apt-get update && apt-get install -y --no-install-recommends \
     ffmpeg \
-    && rm -rf /var/lib/apt/lists/*
+    && apt-get clean && rm -rf /var/lib/apt/lists/* # Clean up apt cache
 
 COPY requirements.txt requirements.txt
+# Use --no-cache-dir during pip install to reduce layer size
 RUN pip install --no-cache-dir --upgrade pip && \
     pip install --no-cache-dir -r requirements.txt