narugo's picture
dev(narugo): more models added
3be3994

A newer version of the Gradio SDK is available: 5.6.0

Upgrade
metadata
title: WD Tagger Heatmaps
emoji: 📚
colorFrom: pink
colorTo: indigo
sdk: gradio
sdk_version: 4.26.0
app_file: app.py
pinned: true
license: mit
preload_from_hub:
  - >-
    SmilingWolf/wd-convnext-tagger-v3
    model.safetensors,config.json,sw_jax_cv_config.json,selected_tags.csv
  - >-
    SmilingWolf/wd-swinv2-tagger-v3
    model.safetensors,config.json,sw_jax_cv_config.json,selected_tags.csv
  - >-
    SmilingWolf/wd-vit-tagger-v3
    model.safetensors,config.json,sw_jax_cv_config.json,selected_tags.csv
  - >-
    SmilingWolf/wd-vit-large-tagger-v3
    model.safetensors,config.json,sw_jax_cv_config.json,selected_tags.csv
  - >-
    SmilingWolf/wd-eva02-large-tagger-v3
    model.safetensors,config.json,sw_jax_cv_config.json,selected_tags.csv
models:
  - SmilingWolf/wd-convnext-tagger-v3
  - SmilingWolf/wd-swinv2-tagger-v3
  - SmilingWolf/wd-vit-tagger-v3
  - SmilingWolf/wd-vit-large-tagger-v3
  - SmilingWolf/wd-eva02-large-tagger-v3

wd-tagger-heatmap

This Space generates heatmaps from the WD Tagger v3 ViT model, showing which parts of the image contributed to the score of a particular tag.