Update Dockerfile

Dockerfile

@@ -1,6 +1,9 @@
+# Dockerfile - Improved with more efficient caching and smaller image size
 FROM python:3.10-slim
 
 WORKDIR /app
+
+# Set environment variables
 ENV HF_HOME=/tmp/huggingface
 ENV TRANSFORMERS_CACHE=/tmp/huggingface/transformers
 ENV HF_DATASETS_CACHE=/tmp/huggingface/datasets
@@ -10,8 +13,8 @@ ENV MPLCONFIGDIR=/tmp/matplotlib
 ENV CUDA_VISIBLE_DEVICES=""
 ENV HOME=/tmp
 
-# Install system dependencies
-RUN apt-get update && apt-get install -y \
+# Install system dependencies - grouped to reduce layers and image size
+RUN apt-get update && apt-get install -y --no-install-recommends \
     git \
     build-essential \
     libgl1-mesa-glx \
@@ -25,7 +28,7 @@ RUN apt-get update && apt-get install -y \
     libopencv-dev \
     && apt-get clean && rm -rf /var/lib/apt/lists/*
 
-# Install Point-E and dependencies
+# Install Point-E and dependencies - ordered from least to most likely to change
 RUN pip install --no-cache-dir -U pip && \
     pip install --no-cache-dir torch==1.13.1 torchvision==0.14.1 --index-url https://download.pytorch.org/whl/cpu && \
     pip install --no-cache-dir flask flask-cors open3d opencv-python-headless requests tqdm && \
@@ -36,8 +39,8 @@ RUN pip install --no-cache-dir -U pip && \
 RUN mkdir -p /tmp/point_e_models
 
 # Copy application code and download script
-COPY app.py .
 COPY download_weights.py .
+COPY app.py .
 
 # Run the download script as part of the container build
 RUN python download_weights.py
@@ -45,5 +48,9 @@ RUN python download_weights.py
 # Expose the port for the Flask server
 EXPOSE 7860
 
+# Add healthcheck
+HEALTHCHECK --interval=30s --timeout=30s --start-period=5s --retries=3 \
+  CMD curl -f http://localhost:7860/health || exit 1
+
 # Command to run the application
 CMD ["python", "app.py"]