WebashalarForML's picture
Update Dockerfile
371810f verified
# Use an official Python runtime as a parent image
FROM python:3.10-slim
# Set environment variables
ENV PYTHONDONTWRITEBYTECODE 1
ENV PYTHONUNBUFFERED 1
ENV HF_HOME=/tmp/huggingface
ENV TF_ENABLE_ONEDNN_OPTS=0
# Set the working directory in the container
WORKDIR /app
# Install system dependencies for OpenCV and other tools
RUN apt-get update && apt-get install -y \
libgl1 \
libglib2.0-0 \
&& rm -rf /var/lib/apt/lists/*
# Copy the requirements file into the container
COPY requirements.txt .
# Install dependencies
RUN pip install --no-cache-dir -r requirements.txt
# Copy the rest of the application code
COPY . .
# Create necessary directories and set permissions for HF Spaces
RUN mkdir -p uploads results encoder && chmod -R 777 uploads results encoder
# Expose the target port
EXPOSE 7860
# Run the application with Gunicorn
# Timeout is set to 0 to disable it, allowing long-running inference tasks
CMD ["gunicorn", "--bind", "0.0.0.0:7860", "--timeout", "0", "app:app"]