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: yellow
 colorTo: indigo
 sdk: gradio
-sdk_version: 3.19.1
+sdk_version: 3.36.1
 app_file: app.py
 pinned: false
 ---

app.py

@@ -2,20 +2,9 @@
 
 from __future__ import annotations
 
-import os
 import pathlib
-import shlex
-import subprocess
-import tarfile
-
-if os.environ.get('SYSTEM') == 'spaces':
-    subprocess.call(shlex.split('pip uninstall -y opencv-python'))
-    subprocess.call(shlex.split('pip uninstall -y opencv-python-headless'))
-    subprocess.call(
-        shlex.split('pip install opencv-python-headless==4.5.5.64'))
 
 import gradio as gr
-import huggingface_hub
 import mediapipe as mp
 import numpy as np
 
@@ -26,24 +15,6 @@ mp_face_mesh = mp.solutions.face_mesh
 TITLE = 'MediaPipe Face Mesh'
 DESCRIPTION = 'https://google.github.io/mediapipe/'
 
-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', '005.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'))
-
 
 def run(
     image: np.ndarray,
@@ -91,9 +62,8 @@ def run(
     return res[:, :, ::-1]
 
 
-image_paths = load_sample_images()
-examples = [[path.as_posix(), 5, 0.5, True, True, True]
-            for path in image_paths]
+image_paths = sorted(pathlib.Path('images').rglob('*.jpg'))
+examples = [[path, 5, 0.5, True, True, True] for path in image_paths]
 
 gr.Interface(
     fn=run,
@@ -113,8 +83,8 @@ gr.Interface(
         gr.Checkbox(label='Show Contours', value=True),
         gr.Checkbox(label='Show Irises', value=True),
     ],
-    outputs=gr.Image(label='Output', type='numpy'),
+    outputs=gr.Image(label='Output', height=500),
     examples=examples,
     title=TITLE,
     description=DESCRIPTION,
-).launch(show_api=False)
+).queue().launch()

requirements.txt

@@ -1,3 +1,3 @@
-mediapipe==0.8.9.1
-numpy==1.22.3
-opencv-python-headless==4.5.5.64
+mediapipe==0.10.1
+numpy==1.23.5
+opencv-python-headless==4.8.0.74