Fix.

2023-09-28 19:45:51 +09:00
parent ae74852536
commit 0b8bec1ac6
2 changed files with 4 additions and 6 deletions
@@ -1,6 +1,6 @@


-from tagger.interrogator import WaifuDiffusionInterrogator
+from tagger.interrogator import Interrogator, WaifuDiffusionInterrogator
 from PIL import Image

 interrogator = WaifuDiffusionInterrogator(
@@ -11,11 +11,10 @@ interrogator = WaifuDiffusionInterrogator(

 im = Image.open("image.jpg")
 result = interrogator.interrogate(im)
+tags = Interrogator.postprocess_tags(result[1])
 print()

-for k, v in result[1].items():
-    if v > 0.25:
-        print(k, v)
-
+for k, v in tags.items():
+    print(k, v)
 print(result[0])

@@ -1,5 +1,4 @@
 import os
-import gc
 import pandas as pd
 import numpy as np