diff --git a/clipseg_rd64_refined/Dockerfile b/clipseg_rd64_refined/Dockerfile new file mode 100644 index 0000000..d525920 --- /dev/null +++ b/clipseg_rd64_refined/Dockerfile @@ -0,0 +1,25 @@ +# please visit https://github.com/xfyun/aiges/releases to get stable and suitable iamges. + +FROM docker.io/library/python:3.8.9 + + +RUN sed -i 's/deb.debian.org/mirrors.ustc.edu.cn/g' /etc/apt/sources.list +RUN sed -i 's|security.debian.org/debian-security|mirrors.ustc.edu.cn/debian-security|g' /etc/apt/sources.list + + +WORKDIR /home/user/app +RUN useradd -m -u 1000 user +RUN chown -R 1000.1000 /home/user + + +RUN pip config set global.index-url https://pypi.mirrors.ustc.edu.cn/simple/ + + +RUN pip install --no-cache-dir pip==22.3.1 +RUN --mount=target=requirements.txt,source=requirements.txt pip install --no-cache-dir -r requirements.txt + + +COPY --chown=1000 ./ /home/user/app + + +CMD ["python3", "app.py"] \ No newline at end of file diff --git a/clipseg_rd64_refined/app.py b/clipseg_rd64_refined/app.py index f7e96a2..b78dd56 100644 --- a/clipseg_rd64_refined/app.py +++ b/clipseg_rd64_refined/app.py @@ -1,9 +1,6 @@ import gradio as gr - - from transformers import AutoProcessor, CLIPSegForImageSegmentation -from PIL import Image -import requests + def inference(img): model_path = "clipseg-rd64-refined" diff --git a/clipseg_rd64_refined/requirements.txt b/clipseg_rd64_refined/requirements.txt new file mode 100644 index 0000000..973b15c --- /dev/null +++ b/clipseg_rd64_refined/requirements.txt @@ -0,0 +1,2 @@ +gradio==3.21.0 +transformers==4.27.1