4K_Upscaler / Dockerfile
Malaji71's picture
Update Dockerfile
d188fd1 verified
# Alternative: Use ubuntu base for better OpenCV compatibility
# FROM ubuntu:22.04
# RUN apt-get update && apt-get install -y python3.10 python3-pip
FROM python:3.10-slim
# Set working directory
WORKDIR /app
# Install system dependencies with updated OpenGL packages
RUN apt-get update && apt-get install -y \
libgl1-mesa-dev \
libglib2.0-0 \
libsm6 \
libxext6 \
libxrender-dev \
libgomp1 \
libgtk-3-0 \
libgstreamer1.0-0 \
libgstreamer-plugins-base1.0-0 \
libavcodec-dev \
libavformat-dev \
libswscale-dev \
ffmpeg \
&& rm -rf /var/lib/apt/lists/*
# Copy requirements first for better caching
COPY requirements.txt .
# Install Python dependencies
RUN pip install --no-cache-dir -r requirements.txt
# Copy application files
COPY . .
# Create necessary directories
RUN mkdir -p /data/uploads /data/outputs
# Set environment variables
ENV PYTHONUNBUFFERED=1
ENV FLASK_APP=app.py
ENV FLASK_ENV=production
# Expose port
EXPOSE 7860
# Run the application
CMD ["python", "app.py"]