fix: make huggingface space happy

Dockerfile

@@ -1,7 +1,20 @@
 FROM docker.io/jeffersonqin/yuzumarker.fontdetection.huggingfacespace.base:latest
 
-WORKDIR /workspace
+RUN useradd -m -u 1000 user
+USER user
+ENV HOME=/home/user \
+	PATH=/home/user/.local/bin:$PATH
 
-COPY . .
+WORKDIR $HOME/app
+USER root
+RUN mv /workspace/font_demo_cache.bin $HOME/app/font_demo_cache.bin
+RUN mv /workspace/demo_fonts $HOME/app/demo_fonts
 
-CMD ["python", "demo.py", "-d", "-1", "-c", "huggingface://gyrojeff/YuzuMarker.FontDetection/commit=bc0f7fc-epoch=26-step=261954.ckpt", "-m", "resnet50", "-z", "512", "-p", "7860"]
+USER user
+COPY --chown=user detector $HOME/app/detector
+COPY --chown=user font_dataset $HOME/app/font_dataset
+COPY --chown=user utils $HOME/app/utils
+COPY --chown=user configs $HOME/app/configs
+COPY --chown=user demo.py $HOME/app/demo.py
+
+CMD ["python", "demo.py", "-d", "-1", "-c", "huggingface://gyrojeff/YuzuMarker.FontDetection/commit=bc0f7fc-epoch=26-step=261954.ckpt", "-m", "resnet50", "-z", "512", "-p", "7860", "-a", "0.0.0.0"]

demo.py

@@ -58,6 +58,13 @@ parser.add_argument(
     default=7860,
     help="Port to use for Gradio (default: 7860)",
 )
+parser.add_argument(
+    "-a",
+    "--address",
+    type=str,
+    default="127.0.0.1",
+    help="Address to use for Gradio (default: 127.0.0.1)",
+)
 
 args = parser.parse_args()
 
@@ -84,6 +91,7 @@ else:
 
 if torch.__version__ >= "2.0" and os.name == "posix":
     model = torch.compile(model)
+    torch._dynamo.config.suppress_errors = True
 
 
 if str(args.checkpoint).startswith("huggingface://"):
@@ -92,7 +100,6 @@ if str(args.checkpoint).startswith("huggingface://"):
     repo = f"{user}/{repo}"
     args.checkpoint = hf_hub_download(repo, file)
 
-
 detector = FontDetector(
     model=model,
     lambda_font=1,
@@ -205,4 +212,4 @@ with gr.Blocks() as demo:
     )
 
 
-demo.launch(share=args.share, server_port=args.port)
+demo.launch(share=args.share, server_port=args.port, server_name=args.address)