fix: docker 环境部署

Dockerfile

@@ -1,6 +1,6 @@
+# https://huggingface.co/spaces/gaunernst/layoutlm-docvqa-paddleocr/blob/main/Dockerfile
 FROM python:3.10
 
-# FROM python:3.10-slim-bullseye
 ENV CUDA_VISIBLE_DEVICES=-1
 ARG PIP_NO_CACHE_DIR=1
 
@@ -16,9 +16,10 @@ RUN pip install paddleocr==2.7.0.3 \
     && pip uninstall -y opencv-python opencv-contrib-python \
     && pip install opencv-python-headless
 
-RUN mkdir /app
-COPY app.py /app
-COPY example_imgs/ /app/example_imgs/
-WORKDIR /app
+RUN useradd -m -u 1000 user
+USER user
+ENV HOME=/home/user
+WORKDIR $HOME/app
+COPY --chown=user . $HOME/app
 
 CMD ["python", "-u", "app.py"]

app.py

@@ -53,18 +53,19 @@ examples = [
 
 css = ".output_image, .input_image {height: 40rem !important; width: 100% !important;}"
 
-demo = gr.Interface(
-    inference,
-    [gr.Image(type='pil', label='Input'),
-     gr.Dropdown(choices=['ch', 'en', 'fr', 'german', 'korean', 'japan'], value='ch', label='language'),
-     gr.Slider(0.1, 1, 0.5, step=0.1, label='confidence_threshold')
-    ],
-    # 输出
-    [gr.Image(type='pil', label='Output'), gr.Dataframe(headers=[ 'bbox', 'score', 'text'], label='Result')],
-    title=title,
-    description=description,
-    examples=examples,
-    css=css,
+if __name__ == '__main__':
+    demo = gr.Interface(
+        inference,
+        [gr.Image(type='pil', label='Input'),
+        gr.Dropdown(choices=['ch', 'en', 'fr', 'german', 'korean', 'japan'], value='ch', label='language'),
+        gr.Slider(0.1, 1, 0.5, step=0.1, label='confidence_threshold')
+        ],
+        # 输出
+        [gr.Image(type='pil', label='Output'), gr.Dataframe(headers=[ 'bbox', 'score', 'text'], label='Result')],
+        title=title,
+        description=description,
+        examples=examples,
+        css=css,
     )
-demo.queue(max_size=10)
-demo.launch(debug=True, share=True)
+    demo.queue(max_size=10)
+    demo.launch(debug=True, server_name="0.0.0.0")

packages.txt

@@ -1,3 +0,0 @@
-ffmpeg 
-libsm6 
-libxext6