diff --git a/clipseg_rd64_refined/Dockerfile b/clipseg_rd64_refined/Dockerfile index a853b01..9572803 100644 --- a/clipseg_rd64_refined/Dockerfile +++ b/clipseg_rd64_refined/Dockerfile @@ -2,6 +2,11 @@ FROM public.ecr.aws/iflytek-open/cuda-go-python-base:11.6-1.17-3.9.13-ubuntu1804 +RUN sed -i 's@//.*archive.ubuntu.com@//mirrors.ustc.edu.cn@g' /etc/apt/sources.list +RUN sed -i 's/security.ubuntu.com/mirrors.ustc.edu.cn/g' /etc/apt/sources.list + +RUN --mount=target=/root/packages.txt,source=packages.txt apt-get update && xargs -r -a /root/packages.txt apt-get install -y && rm -rf /var/lib/apt/lists/* + WORKDIR /home/user/app RUN useradd -m -u 1000 user RUN chown -R 1000.1000 /home/user @@ -13,8 +18,11 @@ 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 +RUN curl -s https://packagecloud.io/install/repositories/github/git-lfs/script.deb.sh | bash +RUN apt-get install git-lfs +RUN git lfs install +RUN git clone http://172.16.59.16:3000/CIDAS/clipseg-rd64-refined.git 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 b78dd56..9847886 100644 --- a/clipseg_rd64_refined/app.py +++ b/clipseg_rd64_refined/app.py @@ -22,7 +22,7 @@ with gr.Blocks() as demo: gr.Markdown( """ # Semantic segmentation:clipseg-rd64-refined - Gradio Demo for clipseg-rd64-refined. To use it, simply upload your image, or click one of the examples to load them. + 这是clipseg-rd64-refined的Gradio Demo,用于语义分割。 """) with gr.Row(): image_input = gr.Image(type="pil") @@ -31,4 +31,4 @@ with gr.Blocks() as demo: image_button.click(inference, inputs=image_input, outputs=text_output) gr.Examples(examples,inputs=image_input) -demo.launch() \ No newline at end of file +demo.launch(server_name="0.0.0.0") \ No newline at end of file diff --git a/clipseg_rd64_refined/clipseg-rd64-refined b/clipseg_rd64_refined/clipseg-rd64-refined deleted file mode 160000 index 583b388..0000000 --- a/clipseg_rd64_refined/clipseg-rd64-refined +++ /dev/null @@ -1 +0,0 @@ -Subproject commit 583b388deb98a04feb3e1f816dcdb8f3062ee205 diff --git a/clipseg_rd64_refined/packages.txt b/clipseg_rd64_refined/packages.txt new file mode 100644 index 0000000..b399c93 --- /dev/null +++ b/clipseg_rd64_refined/packages.txt @@ -0,0 +1,2 @@ +git +curl \ No newline at end of file diff --git a/clipseg_rd64_refined/requirements.txt b/clipseg_rd64_refined/requirements.txt index 973b15c..5e5bf22 100644 --- a/clipseg_rd64_refined/requirements.txt +++ b/clipseg_rd64_refined/requirements.txt @@ -1,2 +1,3 @@ gradio==3.21.0 transformers==4.27.1 +torch==2.0.0 \ No newline at end of file