File size: 1,013 Bytes
0ca4d5d
 
a5b1307
 
 
0ca4d5d
a5b1307
0ca4d5d
 
 
 
a5b1307
 
2e4a3a4
 
4a77736
a5b1307
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
---
title: DanbooruTagQuery
emoji: 🏷️
colorFrom: indigo
colorTo: purple
sdk: gradio
sdk_version: 5.23.3
app_file: app.py
pinned: false
---

# 🏷️ DanbooruTagQuery

ONNX-powered anime image tagger. Upload an image or paste a URL to get predicted Danbooru tags with scores and categories.

Runs **CPU-only** by default. All logic (predictor, preprocessing, category, UI, hub download, model switcher) lives in the shared `danbooru-deploy` package — this `app.py` only imports its `main()` entry point (see [realphongha/DanbooruTagQuery](https://github.com/realphongha/DanbooruTagQuery) `deploy/`).

## Local testing

```bash
# Use a local model file
python app.py /path/to/DanbooruTagQuery_b16_448x448.onnx

# Or point to a directory with .onnx + sidecar files
export MODEL_DIR=/path/to/models/
python app.py
```

## Credits

- Model: [realphongha/danbooru-tag-query](https://huggingface.co/realphongha/danbooru-tag-query)
- Dataset: [danbooru-tags](https://huggingface.co/datasets/qdlabs/danbooru-tags)