quote-caster / Dockerfile
Ubuntu
Fix: set cache dir for Transformers to work on Spaces
429de13
raw
history blame contribute delete
882 Bytes
FROM python:3.10-slim
# Set environment variables
ENV TRANSFORMERS_CACHE=/data/transformers
# Install system dependencies
RUN apt-get update && apt-get install -y git && rm -rf /var/lib/apt/lists/*
# Create writable cache directory
RUN mkdir -p /data/transformers
# Set working directory
WORKDIR /app
# Copy project files
COPY . .
# Install Python packages
RUN pip install --no-cache-dir -r requirements.txt
# Download Hugging Face model at build time (optional but recommended for faster app startup)
RUN python -c "from transformers import AutoTokenizer, AutoModel; \
AutoTokenizer.from_pretrained('aNameNobodyChose/quote-caster-encoder', cache_dir='/data/transformers'); \
AutoModel.from_pretrained('aNameNobodyChose/quote-caster-encoder', cache_dir='/data/transformers')"
# Expose Gradio's default port
EXPOSE 7860
# Run the Gradio app
CMD ["python", "app.py"]