From 93db86fa70b73f9cc459c5f491ba06de3170b6ae Mon Sep 17 00:00:00 2001
From: pharmapsychotic <me@pharmapsychotic.com>
Date: Sun, 5 Feb 2023 12:31:12 -0600
Subject: [PATCH] .

---
 README.md                              | 2 +-
 clip_interrogator/__init__.py          | 2 +-
 clip_interrogator/clip_interrogator.py | 2 +-
 setup.py                               | 2 +-
 4 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/README.md b/README.md
index ff1e3bd..cf266bf 100644
--- a/README.md
+++ b/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
diff --git a/clip_interrogator/__init__.py b/clip_interrogator/__init__.py
index bc18e62..faa94ea 100644
--- a/clip_interrogator/__init__.py
+++ b/clip_interrogator/__init__.py
@@ -1,4 +1,4 @@
 from .clip_interrogator import Interrogator, Config
 
-__version__ = '0.4.2'
+__version__ = '0.4.3'
 __author__ = 'pharmapsychotic'
\ No newline at end of file
diff --git a/clip_interrogator/clip_interrogator.py b/clip_interrogator/clip_interrogator.py
index bde295f..97d32e8 100644
--- a/clip_interrogator/clip_interrogator.py
+++ b/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)
diff --git a/setup.py b/setup.py
index e862dc4..fc429f2 100644
--- a/setup.py
+++ b/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',