Update

.gitattributes

@@ -26,3 +26,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zstandard filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+*.jpg filter=lfs diff=lfs merge=lfs -text

.gitignore

@@ -1 +0,0 @@
-images

README.md

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

app.py

@@ -4,35 +4,13 @@ from __future__ import annotations
 
 import functools
 import os
-import pathlib
-import tarfile
 
 import cv2
 import gradio as gr
-import huggingface_hub
 import numpy as np
 import onnxruntime as ort
 
-TITLE = 'atksh/onnx-facial-lmk-detector'
-DESCRIPTION = 'This is an unofficial demo for https://github.com/atksh/onnx-facial-lmk-detector.'
-
-HF_TOKEN = os.getenv('HF_TOKEN')
-
-
-def load_sample_images() -> list[pathlib.Path]:
-    image_dir = pathlib.Path('images')
-    if not image_dir.exists():
-        image_dir.mkdir()
-        dataset_repo = 'hysts/input-images'
-        filenames = ['001.tar']
-        for name in filenames:
-            path = huggingface_hub.hf_hub_download(dataset_repo,
-                                                   name,
-                                                   repo_type='dataset',
-                                                   use_auth_token=HF_TOKEN)
-            with tarfile.open(path) as f:
-                f.extractall(image_dir.as_posix())
-    return sorted(image_dir.rglob('*.jpg'))
+DESCRIPTION = '# [atksh/onnx-facial-lmk-detector](https://github.com/atksh/onnx-facial-lmk-detector)'
 
 
 def run(image: np.ndarray, sess: ort.InferenceSession) -> np.ndarray:
@@ -58,20 +36,27 @@ sess = ort.InferenceSession('onnx-facial-lmk-detector/model.onnx',
                             sess_options=options,
                             providers=['CPUExecutionProvider'])
 
-func = functools.partial(run, sess=sess)
-
-image_paths = load_sample_images()
-examples = [['onnx-facial-lmk-detector/input.jpg']] + [[path.as_posix()]
-                                                       for path in image_paths]
-
-gr.Interface(
-    fn=func,
-    inputs=gr.Image(label='Input', type='numpy'),
-    outputs=[
-        gr.Image(label='Output', type='numpy'),
-        gr.Gallery(label='Aligned Faces', type='numpy'),
-    ],
-    examples=examples,
-    title=TITLE,
-    description=DESCRIPTION,
-).queue().launch(show_api=False)
+fn = functools.partial(run, sess=sess)
+
+examples = [['onnx-facial-lmk-detector/input.jpg'],
+            ['images/pexels-ksenia-chernaya-8535230.jpg']]
+
+with gr.Blocks(css='style.css') as demo:
+    gr.Markdown(DESCRIPTION)
+    with gr.Row():
+        with gr.Column():
+            image = gr.Image(label='Input', type='numpy')
+            run_button = gr.Button()
+        with gr.Column():
+            result = gr.Image(label='Output')
+            gallery = gr.Gallery(label='Aligned Faces')
+    gr.Examples(examples=examples,
+                inputs=image,
+                outputs=[result, gallery],
+                fn=fn,
+                cache_examples=os.getenv('CACHE_EXAMPLES') == '1')
+    run_button.click(fn=fn,
+                     inputs=image,
+                     outputs=[result, gallery],
+                     api_name='run')
+demo.queue(max_size=10).launch()

requirements.txt

@@ -1,3 +1,3 @@
-numpy==1.22.3
-onnxruntime==1.11.0
-opencv-python-headless==4.5.5.64
+numpy==1.23.5
+onnxruntime==1.15.0
+opencv-python-headless==4.8.0.74

style.css

@@ -0,0 +1,7 @@
+h1 {
+  text-align: center;
+}
+
+#duplicate-button {
+  margin: auto;
+}