Update Dockerfile

Dockerfile

@@ -1,6 +1,6 @@
-FROM python:3.10-slim
+FROM python:3.10-slim as builder
 
-# Minimal system dependencies
+# System dependencies
 RUN apt-get update && apt-get install -y \
     git \
     git-lfs \
@@ -10,11 +10,32 @@ RUN apt-get update && apt-get install -y \
 WORKDIR /app
 COPY . .
 
-# Install Python dependencies with memory optimizations
+# Install with separate layers for better caching
+RUN pip install --no-cache-dir torch==2.2.0 --index-url https://download.pytorch.org/whl/cpu
 RUN pip install --no-cache-dir \
-    torch==2.2.0 \
     transformers==4.40.0 \
     gradio==4.24.0 \
     accelerate==0.29.0
 
+# Runtime image
+FROM python:3.10-slim
+WORKDIR /app
+
+# Copy only necessary files
+COPY --from=builder /usr/local/lib/python3.10/site-packages /usr/local/lib/python3.10/site-packages
+COPY --from=builder /usr/local/bin/git-lfs /usr/local/bin/
+COPY --from=builder /app/app.py .
+
+# Environment variables
+ENV PYTHONUNBUFFERED=1 \
+    GRADIO_SERVER_NAME=0.0.0.0 \
+    GRADIO_SERVER_PORT=7860 \
+    HF_TOKEN="" \
+    DEVICE="cpu"
+
+# Health check
+HEALTHCHECK --interval=30s --timeout=30s --start-period=5s --retries=3 \
+    CMD curl -f http://localhost:7860 || exit 1
+
+EXPOSE 7860
 CMD ["python", "app.py"]