.

README.md

@@ -36,7 +36,7 @@ Install with PIP
 pip3 install torch torchvision --extra-index-url https://download.pytorch.org/whl/cu117
 
 # install clip-interrogator
-pip install clip-interrogator==0.4.2
+pip install clip-interrogator==0.4.3
 ```
 
 You can then use it in your script

clip_interrogator/__init__.py

@@ -1,4 +1,4 @@
 from .clip_interrogator import Interrogator, Config
 
-__version__ = '0.4.2'
+__version__ = '0.4.3'
 __author__ = 'pharmapsychotic'

clip_interrogator/clip_interrogator.py

@@ -271,7 +271,7 @@ class Interrogator():
         merged = _merge_tables([self.artists, self.flavors, self.mediums, self.movements, self.trendings], self.config)
         flaves = merged.rank(image_features, self.config.flavor_intermediate_count)
 
-        best_prompt, best_sim = caption, self.similarity(image_features, best_prompt)
+        best_prompt, best_sim = caption, self.similarity(image_features, caption)
         best_prompt = self.chain(image_features, flaves, best_prompt, best_sim, min_count=min_flavors, max_count=max_flavors, desc="Flavor chain")
 
         fast_prompt = self.interrogate_fast(image, max_flavors)

setup.py

@@ -5,7 +5,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="clip-interrogator",
-    version="0.4.2",
+    version="0.4.3",
     license='MIT',
     author='pharmapsychotic',
     author_email='me@pharmapsychotic.com',