diff --git a/maskformer_swin_large_coco/Dockerfile b/maskformer_swin_large_coco/Dockerfile index 2f124b8..c5db357 100644 --- a/maskformer_swin_large_coco/Dockerfile +++ b/maskformer_swin_large_coco/Dockerfile @@ -2,6 +2,10 @@ 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 @@ -14,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/facebook/maskformer-swin-large-coco.git CMD ["python3", "app.py"] \ No newline at end of file diff --git a/maskformer_swin_large_coco/app.py b/maskformer_swin_large_coco/app.py index 720e382..69d5347 100644 --- a/maskformer_swin_large_coco/app.py +++ b/maskformer_swin_large_coco/app.py @@ -27,7 +27,7 @@ with gr.Blocks() as demo: gr.Markdown( """ # Panorama segmentation:maskformer-swin-large-coco - Gradio Demo for maskformer-swin-large-coco. To use it, simply upload your image, or click one of the examples to load them. + 这是maskformer-swin-large-coco的Gradio Demo,用于全景分割。上传你想要的图像或者点击下面的示例来加载它。 """) with gr.Row(): image_input = gr.Image(type="pil") @@ -36,7 +36,7 @@ with gr.Blocks() as demo: image_button.click(inference, inputs=image_input, outputs=text_output) gr.Examples(examples,inputs=image_input) -demo.launch() +demo.launch(server_name="0.0.0.0") diff --git a/maskformer_swin_large_coco/maskformer-swin-large-coco b/maskformer_swin_large_coco/maskformer-swin-large-coco deleted file mode 160000 index 5f009df..0000000 --- a/maskformer_swin_large_coco/maskformer-swin-large-coco +++ /dev/null @@ -1 +0,0 @@ -Subproject commit 5f009df2186c5841582ecd419409b56041180144 diff --git a/maskformer_swin_large_coco/packages.txt b/maskformer_swin_large_coco/packages.txt new file mode 100644 index 0000000..b399c93 --- /dev/null +++ b/maskformer_swin_large_coco/packages.txt @@ -0,0 +1,2 @@ +git +curl \ No newline at end of file diff --git a/maskformer_swin_large_coco/requirements.txt b/maskformer_swin_large_coco/requirements.txt index 973b15c..3244b33 100644 --- a/maskformer_swin_large_coco/requirements.txt +++ b/maskformer_swin_large_coco/requirements.txt @@ -1,2 +1,3 @@ gradio==3.21.0 transformers==4.27.1 +torch==2.0.0