We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent de1528f commit 51188b7Copy full SHA for 51188b7
turftopic/multimodal.py
@@ -216,7 +216,7 @@ def collect_top_images(
216
if negative:
217
image_topic_vector = -image_topic_vector
218
top_im_ind = np.argsort(-image_topic_vector)[:20]
219
- top_im = [images[i] for i in top_im_ind]
+ top_im = [images[int(i)] for i in top_im_ind]
220
top_images.append(top_im)
221
return top_images
222
0 commit comments