text2tags
Text2tags is a Python library that uses a finetune of Llama-7b and llama cpp to predict Danbooru tags from natural text.
Installation
Use the package manager [pip]
pip install text2tags-lib
Usage
from text2tags import TaggerLlama
model = TaggerLlama()
tags = model.predict_tags("Minato Aqua from hololive with pink and blue twintails in a blue maid outfit")
print(', '.join(tags))
Demos
Other links
License
Release files for text2tags-lib 0.0.7
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| text2tags-lib-0.0.7.tar.gz | 75.4 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| text2tags_lib-0.0.7-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 149.7 kB