| FROM python:3.10-slim | |
| WORKDIR /app | |
| # Install system dependencies | |
| RUN apt-get update && apt-get install -y \ | |
| git \ | |
| git-lfs \ | |
| build-essential \ | |
| && rm -rf /var/lib/apt/lists/* | |
| # Initialize git-lfs | |
| RUN git lfs install | |
| # Install Python packages | |
| COPY requirements.txt . | |
| RUN pip install --no-cache-dir --upgrade pip && \ | |
| pip install --no-cache-dir -r requirements.txt | |
| # Copy application files | |
| COPY . . | |
| # Set up Hugging Face cache directory | |
| ENV HF_HOME=/app/cache | |
| ENV TRANSFORMERS_CACHE=/app/cache | |
| ENV HF_HUB_CACHE=/app/cache | |
| # Set environment variables for better performance | |
| ENV PYTORCH_CUDA_ALLOC_CONF=max_split_size_mb:512 | |
| ENV CUDA_VISIBLE_DEVICES=0,1,2,3 | |
| # Expose port for the application | |
| EXPOSE 7860 | |
| # Run the application | |
| CMD ["python", "app.py"] |