update

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"]

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")
 
 
 

maskformer_swin_large_coco/maskformer-swin-large-coco

@@ -1 +0,0 @@
-Subproject commit 5f009df2186c5841582ecd419409b56041180144

maskformer_swin_large_coco/packages.txt

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

maskformer_swin_large_coco/requirements.txt

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