From 9f06fadd88c2d8637a42d21b971c842a95d60da9 Mon Sep 17 00:00:00 2001
From: SOULOFCINDER <841135647@qq.com>
Date: Fri, 31 Mar 2023 15:41:41 +0800
Subject: [PATCH] update

---
 maskformer_swin_large_coco/Dockerfile       | 25 +++++++++++++++++++++
 maskformer_swin_large_coco/app.py           |  2 +-
 maskformer_swin_large_coco/requirements.txt |  2 ++
 3 files changed, 28 insertions(+), 1 deletion(-)
 create mode 100644 maskformer_swin_large_coco/Dockerfile
 create mode 100644 maskformer_swin_large_coco/requirements.txt

diff --git a/maskformer_swin_large_coco/Dockerfile b/maskformer_swin_large_coco/Dockerfile
new file mode 100644
index 0000000..d525920
--- /dev/null
+++ b/maskformer_swin_large_coco/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/maskformer_swin_large_coco/app.py b/maskformer_swin_large_coco/app.py
index c0f9da4..720e382 100644
--- a/maskformer_swin_large_coco/app.py
+++ b/maskformer_swin_large_coco/app.py
@@ -1,6 +1,6 @@
 import gradio as gr
 from transformers import MaskFormerImageProcessor, MaskFormerForInstanceSegmentation
-from PIL import Image
+
 def inference(img):
     # load MaskFormer fine-tuned on COCO panoptic segmentation
     model_path = "maskformer-swin-large-coco"
diff --git a/maskformer_swin_large_coco/requirements.txt b/maskformer_swin_large_coco/requirements.txt
new file mode 100644
index 0000000..973b15c
--- /dev/null
+++ b/maskformer_swin_large_coco/requirements.txt
@@ -0,0 +1,2 @@
+gradio==3.21.0
+transformers==4.27.1