From 0b8bec1ac669f09e7b04fd72d13e9ce6c1a34db8 Mon Sep 17 00:00:00 2001
From: corkborg <62742115+corkborg@users.noreply.github.com>
Date: Thu, 28 Sep 2023 19:45:51 +0900
Subject: [PATCH] Fix.

---
 run.py                 | 9 ++++-----
 tagger/interrogator.py | 1 -
 2 files changed, 4 insertions(+), 6 deletions(-)

diff --git a/run.py b/run.py
index 1fd781b..deb40c7 100644
--- a/run.py
+++ b/run.py
@@ -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])
 
diff --git a/tagger/interrogator.py b/tagger/interrogator.py
index 4fde4df..cb7dd48 100644
--- a/tagger/interrogator.py
+++ b/tagger/interrogator.py
@@ -1,5 +1,4 @@
 import os
-import gc
 import pandas as pd
 import numpy as np