File size: 1,805 Bytes
74d6c39 | 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 58 59 60 61 62 63 64 65 66 67 | # Use an official Python image as the base image
FROM python:3.10-slim
# Set environment variables to prevent Python from writing .pyc files and to buffer stdout and stderr
ENV PYTHONDONTWRITEBYTECODE 1
ENV PYTHONUNBUFFERED 1
# Set the working directory inside the container
WORKDIR /app
# Install system dependencies
RUN apt-get update && apt-get install -y --no-install-recommends \
build-essential \
wget \
git \
curl \
ca-certificates \
&& rm -rf /var/lib/apt/lists/*
# Install pip and upgrade it
RUN pip install --no-cache-dir --upgrade pip
# Install PyTorch and TorchVision
RUN pip install --no-cache-dir \
torch==2.5.1 \
torchvision==0.20.1 \
-f https://download.pytorch.org/whl/torch_stable.html
# Clone SAM2 repository and install it
RUN git clone https://github.com/facebookresearch/sam2.git sam2_repo && \
cd sam2_repo && \
pip install -e .
# Download SAM2 model checkpoints
RUN cd /app/sam2_repo/checkpoints && \
chmod +x download_ckpts.sh && \
./download_ckpts.sh
RUN apt-get update && apt-get install ffmpeg libsm6 libxext6 -y
# Install Flask application dependencies
RUN pip install --no-cache-dir \
httpx==0.27.2 \
Flask==3.1.0 \
Flask-Cors==5.0.0 \
openai==1.55.3 \
matplotlib==3.9.2 \
tqdm==4.67.0 \
opencv-python==4.10.0.84 \
albumentations==1.4.21 \
gunicorn
# Copy the application code to the container
COPY . /app/
# Download ResNet50 checkpoint
RUN curl https://download.pytorch.org/models/resnet50-0676ba61.pth --output /app/resnet50-0676ba61.pth
# Copy SAM2 checkpoints
RUN mv /app/sam2_repo/checkpoints/*.pt /app/ && ls /app/
# Expose the port the app runs on
EXPOSE 7860
# Define the command to run the Flask app
CMD ["gunicorn", "-b=0.0.0.0:7860", "app:app", "--timeout=0"]
|