Update

README.md

@@ -4,7 +4,7 @@ emoji: 🏢
 colorFrom: gray
 colorTo: purple
 sdk: gradio
-sdk_version: 3.36.1
+sdk_version: 4.36.0
 app_file: app.py
 pinned: false
 ---

app.py

@@ -2,9 +2,6 @@
 
 from __future__ import annotations
 
-import functools
-import os
-
 import cv2
 import gradio as gr
 import numpy as np
@@ -12,12 +9,19 @@ import onnxruntime as ort
 
 DESCRIPTION = "# [atksh/onnx-facial-lmk-detector](https://github.com/atksh/onnx-facial-lmk-detector)"
 
+options = ort.SessionOptions()
+options.intra_op_num_threads = 8
+options.inter_op_num_threads = 8
+sess = ort.InferenceSession(
+    "onnx-facial-lmk-detector/model.onnx", sess_options=options, providers=["CPUExecutionProvider"]
+)
+
 
-def run(image: np.ndarray, sess: ort.InferenceSession) -> np.ndarray:
+def run(image: np.ndarray) -> np.ndarray:
     # float32, int, int, uint8, int, float32
     # (N,), (N, 4), (N, 5, 2), (N, 224, 224, 3), (N, 106, 2), (N, 2, 3)
     scores, bboxes, keypoints, aligned_images, landmarks, affine_matrices = sess.run(
-        None, {"input": image[:, :, ::-1]}
+        None, {"input": image[:, :, ::-1].copy()}
     )
 
     res = image[:, :, ::-1].copy()
@@ -30,16 +34,7 @@ def run(image: np.ndarray, sess: ort.InferenceSession) -> np.ndarray:
     return res[:, :, ::-1], [face[:, :, ::-1] for face in aligned_images]
 
 
-options = ort.SessionOptions()
-options.intra_op_num_threads = 8
-options.inter_op_num_threads = 8
-sess = ort.InferenceSession(
-    "onnx-facial-lmk-detector/model.onnx", sess_options=options, providers=["CPUExecutionProvider"]
-)
-
-fn = functools.partial(run, sess=sess)
-
-examples = [["onnx-facial-lmk-detector/input.jpg"], ["images/pexels-ksenia-chernaya-8535230.jpg"]]
+examples = ["onnx-facial-lmk-detector/input.jpg", "images/pexels-ksenia-chernaya-8535230.jpg"]
 
 with gr.Blocks(css="style.css") as demo:
     gr.Markdown(DESCRIPTION)
@@ -54,8 +49,14 @@ with gr.Blocks(css="style.css") as demo:
         examples=examples,
         inputs=image,
         outputs=[result, gallery],
-        fn=fn,
-        cache_examples=os.getenv("CACHE_EXAMPLES") == "1",
+        fn=run,
     )
-    run_button.click(fn=fn, inputs=image, outputs=[result, gallery], api_name="run")
-demo.queue(max_size=10).launch()
+    run_button.click(
+        fn=run,
+        inputs=image,
+        outputs=[result, gallery],
+        api_name="run",
+    )
+
+if __name__ == "__main__":
+    demo.queue(max_size=10).launch()

requirements.txt

@@ -1,3 +1,3 @@
-numpy==1.23.5
+numpy==1.26.4
 onnxruntime==1.15.0
-opencv-python-headless==4.8.0.74
+opencv-python-headless==4.10.0.82

style.css

@@ -1,7 +1,11 @@
 h1 {
   text-align: center;
+  display: block;
 }
 
 #duplicate-button {
   margin: auto;
+  color: #fff;
+  background: #1565c0;
+  border-radius: 100vh;
 }