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