Skip to content

Commit

Permalink
update README
Browse files Browse the repository at this point in the history
  • Loading branch information
SmashinFries committed Mar 21, 2024
1 parent b327fb9 commit 907f510
Showing 1 changed file with 13 additions and 9 deletions.
22 changes: 13 additions & 9 deletions README.md
Original file line number Diff line number Diff line change
@@ -1,29 +1,33 @@
# WaifuTagger
![](./banner.png)
![banner](./banner.png)

<a href="https://huggingface.co/Smashinfries/wd-convnext-tagger-v3-mobile" target="_blank">
<img alt="Hugging Face" src="https://img.shields.io/badge/%F0%9F%A4%97%20Model%20Card-ffc107?color=ffc107&logoColor=white" />
</a>
<!-- UPDATE WHEN ON PLAY STORE -->
<!-- <a href="https://github.com/KuzuLabz/WaifuTagger/releases/latest" target="_blank">
<img alt="Hugging Face" src="https://img.shields.io/badge/Google Play-Download-Test?logo=googleplay" />
</a> -->
<a href="https://play.google.com/apps/testing/com.kuzulabz.WaifuTagger" target="_blank">
<img alt="Hugging Face" src="https://img.shields.io/badge/Google Play-Join Beta-Test?logo=googleplay" />
</a>
<a href="https://github.com/KuzuLabz/WaifuTagger/releases/latest" target="_blank">
<img alt="Hugging Face" src="https://img.shields.io/badge/Android-Release-Release?logo=android" />
</a>


Run SmilingWolfs [wd-convnext-tagger-v3](https://huggingface.co/SmilingWolf/wd-convnext-tagger-v3) model on mobile!

## Platforms
Android is currently supported - looking to get this on Google Play soon!
## 🎉 Google Play Closed Beta
The app is currently in closed beta and needs testers! You can join the beta [here](https://play.google.com/apps/testing/com.kuzulabz.WaifuTagger) or from the Google Play button above.

*Due to new Google Play policies, 20 testers are needed for 14 days before the app can go public.*

## 🎮 Platforms
Android is currently supported!

I think iOS could easily be supported but I haven't tested it. Same goes for web.

## Model
## 🤗 Model
To run the model on mobile, I quantized it and added preprocessing using [onnxruntime-extentions](https://onnxruntime.ai/docs/extensions/). This model is available on my [HuggingFace](https://huggingface.co/Smashinfries/wd-convnext-tagger-v3-mobile). You can also find more details about the process there.

## Model Accuracy
## 🎯 Model Accuracy
If you get bad results, it's highly likely that I did something wrong with the preprocessing. Any tips for improvement would be appreciated.

## Contributing
Expand Down

0 comments on commit 907f510

Please sign in to comment.