yukee1992's picture
Update Dockerfile
25bfa83 verified
raw
history blame
408 Bytes
FROM python:3.10-slim
# Minimal system dependencies
RUN apt-get update && apt-get install -y \
git \
git-lfs \
&& rm -rf /var/lib/apt/lists/* \
&& git lfs install
WORKDIR /app
COPY . .
# Install Python dependencies with memory optimizations
RUN pip install --no-cache-dir \
torch==2.2.0 \
transformers==4.40.0 \
gradio==4.24.0 \
accelerate==0.29.0
CMD ["python", "app.py"]