emotion-detection-api / Dockerfile
HimAJ's picture
Update Dockerfile
96b63cd verified
# Dockerfile for Hugging Face Spaces
FROM python:3.11-slim
ENV PYTHONUNBUFFERED=1
WORKDIR /app
# System dependencies for opencv and runtime model download
RUN apt-get update && apt-get install -y --no-install-recommends \
build-essential \
libgl1 \
libglib2.0-0 \
curl \
&& rm -rf /var/lib/apt/lists/*
# Copy requirements first (for better caching)
COPY requirements.txt /app/requirements.txt
# Upgrade pip
RUN python -m pip install --upgrade pip setuptools wheel
# Install requirements
RUN pip install --no-cache-dir -r requirements.txt
# Copy all app code
COPY . /app/
# Debug: List what's in scripts directory
RUN ls -la /app/scripts/ 2>/dev/null || echo "Scripts directory not found"
# Handle entrypoint - rename if needed, or create simple one
RUN mkdir -p /app/scripts && \
if [ -f /app/scripts/entrypoint_hf.sh ]; then \
mv /app/scripts/entrypoint_hf.sh /app/scripts/entrypoint.sh; \
elif [ ! -f /app/scripts/entrypoint.sh ]; then \
echo '#!/bin/sh' > /app/scripts/entrypoint.sh && \
echo 'set -eu' >> /app/scripts/entrypoint.sh && \
echo 'ASRIPA_MODEL_DIR="/app/models/fine_tuned_vit"' >> /app/scripts/entrypoint.sh && \
echo 'ASRIPA_MODEL_ID="${ASRIPA_MODEL_ID:-HimAJ/asripa-emotion-detection}"' >> /app/scripts/entrypoint.sh && \
echo 'if [ -n "$ASRIPA_MODEL_ID" ] && [ ! -f "$ASRIPA_MODEL_DIR/model.safetensors" ]; then' >> /app/scripts/entrypoint.sh && \
echo ' echo "📥 Downloading Asripa model from HuggingFace..."' >> /app/scripts/entrypoint.sh && \
echo ' echo " Model ID: $ASRIPA_MODEL_ID"' >> /app/scripts/entrypoint.sh && \
echo ' mkdir -p "$ASRIPA_MODEL_DIR"' >> /app/scripts/entrypoint.sh && \
echo ' python3 -c "from huggingface_hub import snapshot_download; import os, sys; snapshot_download(repo_id=\"$ASRIPA_MODEL_ID\", local_dir=\"$ASRIPA_MODEL_DIR\", local_dir_use_symlinks=False) or sys.exit(0)" || echo "⚠️ Asripa download skipped"' >> /app/scripts/entrypoint.sh && \
echo 'fi' >> /app/scripts/entrypoint.sh && \
echo 'PORT="${PORT:-7860}"' >> /app/scripts/entrypoint.sh && \
echo 'echo "Starting gunicorn on 0.0.0.0:${PORT}"' >> /app/scripts/entrypoint.sh && \
echo 'export PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION=python' >> /app/scripts/entrypoint.sh && \
echo 'exec gunicorn main:app --bind 0.0.0.0:"${PORT}" --workers 1 --threads 1 --timeout 120 --worker-class gthread' >> /app/scripts/entrypoint.sh; \
fi && \
chmod +x /app/scripts/entrypoint.sh
# Verify entrypoint exists
RUN test -f /app/scripts/entrypoint.sh || (echo "ERROR: entrypoint.sh not found!" && exit 1)
# Hugging Face Spaces uses port 7860
EXPOSE 7860
# Use entrypoint script
ENTRYPOINT ["/app/scripts/entrypoint.sh"]