This commit is contained in:
SOULOFCINDER 2023-04-06 17:36:57 +08:00
parent 204c98a67e
commit f1ff743680
5 changed files with 14 additions and 4 deletions

View File

@ -2,6 +2,11 @@
FROM public.ecr.aws/iflytek-open/cuda-go-python-base:11.6-1.17-3.9.13-ubuntu1804 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 WORKDIR /home/user/app
RUN useradd -m -u 1000 user RUN useradd -m -u 1000 user
RUN chown -R 1000.1000 /home/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 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 RUN --mount=target=requirements.txt,source=requirements.txt pip install --no-cache-dir -r requirements.txt
COPY --chown=1000 ./ /home/user/app 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"] CMD ["python3", "app.py"]

View File

@ -22,7 +22,7 @@ with gr.Blocks() as demo:
gr.Markdown( gr.Markdown(
""" """
# Semantic segmentation:clipseg-rd64-refined # 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(): with gr.Row():
image_input = gr.Image(type="pil") 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) image_button.click(inference, inputs=image_input, outputs=text_output)
gr.Examples(examples,inputs=image_input) gr.Examples(examples,inputs=image_input)
demo.launch() demo.launch(server_name="0.0.0.0")

@ -1 +0,0 @@
Subproject commit 583b388deb98a04feb3e1f816dcdb8f3062ee205

View File

@ -0,0 +1,2 @@
git
curl

View File

@ -1,2 +1,3 @@
gradio==3.21.0 gradio==3.21.0
transformers==4.27.1 transformers==4.27.1
torch==2.0.0