Metadata-Version: 2.1
Name: text2tags-lib
Version: 0.0.4
Summary: Use a finetune of Llama-7b and llama cpp to predict Danbooru tags from natural text.
Home-page: https://github.com/DatboiiPuntai/text2tags-lib
Download-URL: https://github.com/DatboiiPuntai/text2tags-lib/archive/refs/heads/master.zip
Author: Pantai Suyasri
Author-email: pantaisuyasri@gmail.com
License: MIT
Description-Content-Type: text/markdown

# 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

```python
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))
```

## License

[MIT](https://choosealicense.com/licenses/mit/)
