Skip to content

Commit 51188b7

Browse files
Fixed indexing error in multimodal models
1 parent de1528f commit 51188b7

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

turftopic/multimodal.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -216,7 +216,7 @@ def collect_top_images(
216216
if negative:
217217
image_topic_vector = -image_topic_vector
218218
top_im_ind = np.argsort(-image_topic_vector)[:20]
219-
top_im = [images[i] for i in top_im_ind]
219+
top_im = [images[int(i)] for i in top_im_ind]
220220
top_images.append(top_im)
221221
return top_images
222222

0 commit comments

Comments
 (0)