From 36f2e71d3bc2bbb0386d8316a02c8ea584f1ec45 Mon Sep 17 00:00:00 2001
From: corkborg <62742115+corkborg@users.noreply.github.com>
Date: Thu, 28 Sep 2023 19:26:51 +0900
Subject: [PATCH] Add envs.

---
 .gitignore       |  6 ++++++
 requirements.txt |  9 +++++++++
 run.py           | 21 +++++++++++++++++++++
 3 files changed, 36 insertions(+)
 create mode 100644 .gitignore
 create mode 100644 requirements.txt
 create mode 100644 run.py

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])
+