TrungTran commited on
Commit
2a9e2ae
·
verified ·
1 Parent(s): 77d52f7

Upload app.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +8 -5
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 = "trungthanhtran/faceage-dino",
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/trungthanhtran/faceage-dino)
277
  """
278
 
279
- with gr.Blocks(title="FaceAge-DINOv3", theme=gr.themes.Soft()) as demo:
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
- *Trained on LAGENDA · IMDB-Clean · UTKFace · AgeDB · FairFace · Open Images.*
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())