Spaces:
Running
Running
Upload app.py with huggingface_hub
Browse files
app.py
CHANGED
|
@@ -6,6 +6,8 @@ Age/gender : FaceAge-DINOv3 ONNX (CPU, ~1.2 GB)
|
|
| 6 |
"""
|
| 7 |
import os
|
| 8 |
import numpy as np
|
|
|
|
|
|
|
| 9 |
import gradio as gr
|
| 10 |
from PIL import Image, ImageDraw, ImageFont
|
| 11 |
|
|
@@ -58,8 +60,9 @@ def _load_age_model():
|
|
| 58 |
|
| 59 |
print("[AgeModel] Downloading ONNX from HuggingFace Hub …")
|
| 60 |
onnx_path = hf_hub_download(
|
| 61 |
-
repo_id = "
|
| 62 |
filename = "faceage_dino_fp32.onnx",
|
|
|
|
| 63 |
)
|
| 64 |
|
| 65 |
opts = ort.SessionOptions()
|
|
@@ -273,10 +276,10 @@ Upload a photo. **YuNet** auto-detects faces, then **FaceAge-DINOv3** predicts a
|
|
| 273 |
| ⚡ Speed | ~100 ms / face on CPU (ONNX FP32) |
|
| 274 |
| 🔍 Detector | YuNet (OpenCV, ~350 KB) |
|
| 275 |
|
| 276 |
-
[📄 Model Card](https://huggingface.co/
|
| 277 |
"""
|
| 278 |
|
| 279 |
-
with gr.Blocks(title="FaceAge-DINOv3"
|
| 280 |
gr.Markdown(_DESC)
|
| 281 |
|
| 282 |
with gr.Row():
|
|
@@ -302,9 +305,9 @@ with gr.Blocks(title="FaceAge-DINOv3", theme=gr.themes.Soft()) as demo:
|
|
| 302 |
|
| 303 |
gr.Markdown("""
|
| 304 |
---
|
| 305 |
-
*
|
| 306 |
*DINOv3-ViT-L pretrained by Meta AI on LVD-1.68B images.*
|
| 307 |
""")
|
| 308 |
|
| 309 |
if __name__ == "__main__":
|
| 310 |
-
demo.launch()
|
|
|
|
| 6 |
"""
|
| 7 |
import os
|
| 8 |
import numpy as np
|
| 9 |
+
|
| 10 |
+
_HF_TOKEN = os.environ.get("HF_TOKEN") or None
|
| 11 |
import gradio as gr
|
| 12 |
from PIL import Image, ImageDraw, ImageFont
|
| 13 |
|
|
|
|
| 60 |
|
| 61 |
print("[AgeModel] Downloading ONNX from HuggingFace Hub …")
|
| 62 |
onnx_path = hf_hub_download(
|
| 63 |
+
repo_id = "TrungTran/faceage-dino",
|
| 64 |
filename = "faceage_dino_fp32.onnx",
|
| 65 |
+
token = _HF_TOKEN,
|
| 66 |
)
|
| 67 |
|
| 68 |
opts = ort.SessionOptions()
|
|
|
|
| 276 |
| ⚡ Speed | ~100 ms / face on CPU (ONNX FP32) |
|
| 277 |
| 🔍 Detector | YuNet (OpenCV, ~350 KB) |
|
| 278 |
|
| 279 |
+
[📄 Model Card](https://huggingface.co/TrungTran/faceage-dino)
|
| 280 |
"""
|
| 281 |
|
| 282 |
+
with gr.Blocks(title="FaceAge-DINOv3") as demo:
|
| 283 |
gr.Markdown(_DESC)
|
| 284 |
|
| 285 |
with gr.Row():
|
|
|
|
| 305 |
|
| 306 |
gr.Markdown("""
|
| 307 |
---
|
| 308 |
+
*Our Collection: LAGENDA · IMDB-Clean · UTKFace · AgeDB · FairFace · Open Images.*
|
| 309 |
*DINOv3-ViT-L pretrained by Meta AI on LVD-1.68B images.*
|
| 310 |
""")
|
| 311 |
|
| 312 |
if __name__ == "__main__":
|
| 313 |
+
demo.launch(theme=gr.themes.Soft())
|