File size: 1,763 Bytes
ca75c2d 7eac98c ca75c2d 7eac98c ca75c2d f5d3311 ca75c2d 7eac98c e716c29 7eac98c ca75c2d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 |
# Dockerfile for Hugging Face Spaces (Docker SDK)
# This is the main Dockerfile used by HF Spaces when sdk: docker is set
FROM python:3.10-slim
# System dependencies (ADD GIT HERE)
RUN apt-get update && apt-get install -y \
gcc \
g++ \
cmake \
libopenblas-dev \
libomp-dev \
curl \
git \
&& rm -rf /var/lib/apt/lists/*
# Set working directory
WORKDIR /app
# Set environment variables to prevent warnings
# Fix libgomp OMP_NUM_THREADS warning
ENV OMP_NUM_THREADS=4
ENV MKL_NUM_THREADS=4
# Copy requirements first for better caching
COPY requirements.txt .
# Install Python dependencies
RUN pip install --no-cache-dir -r requirements.txt
# Copy application code
COPY . .
# Expose ports
# 7860: Gradio UI (primary, exposed by HF Spaces)
# 5001: Flask API (runs in background thread)
EXPOSE 7860 5001
# Health check - check Gradio endpoint
HEALTHCHECK --interval=30s --timeout=30s --start-period=60s --retries=3 \
CMD curl -f http://localhost:7860/ || exit 1
# Run main.py (starts both Gradio and Flask API)
# main.py starts Flask API in background thread and launches Gradio
#
# ZeroGPU API Configuration (optional):
# Set these environment variables in HF Spaces secrets for ZeroGPU integration:
# - USE_ZERO_GPU=true (enable ZeroGPU API)
# - ZERO_GPU_API_URL=https://<pod-id>-8000.proxy.runpod.net (Runpod proxy URL)
# Example: https://bm9njt1ypzvuqw-8000.proxy.runpod.net
# - ZERO_GPU_EMAIL=service@example.com (for service account mode)
# - ZERO_GPU_PASSWORD=your-password (for service account mode)
# - ZERO_GPU_PER_USER_MODE=true (for per-user mode, optional)
# - ZERO_GPU_ADMIN_EMAIL=admin@example.com (for per-user mode)
# - ZERO_GPU_ADMIN_PASSWORD=admin-password (for per-user mode)
CMD ["python", "main.py"]
|