diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..fd6106c --- /dev/null +++ b/.gitignore @@ -0,0 +1,6 @@ +__pycache__/ +.vscode/ +.venv/ +.env + +presets/ \ No newline at end of file diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 0000000..38b01cd --- /dev/null +++ b/requirements.txt @@ -0,0 +1,9 @@ +pandas==2.1.1 +huggingface-hub==0.17.3 +torch==2.0.1 +Pillow==10.0.1 +# tensorflow==2.14.0 +# tensorflow-io==0.34.0 +deepdanbooru==1.0.2 +onnxruntime==1.16.0 +opencv-python==4.8.0.76 \ No newline at end of file diff --git a/run.py b/run.py new file mode 100644 index 0000000..1fd781b --- /dev/null +++ b/run.py @@ -0,0 +1,21 @@ + + +from tagger.interrogator import WaifuDiffusionInterrogator +from PIL import Image + +interrogator = WaifuDiffusionInterrogator( + 'wd14-convnextv2-v2', + repo_id='SmilingWolf/wd-v1-4-convnextv2-tagger-v2', + revision='v2.0' +) + +im = Image.open("image.jpg") +result = interrogator.interrogate(im) +print() + +for k, v in result[1].items(): + if v > 0.25: + print(k, v) + +print(result[0]) +