Upload app.py with huggingface_hub

app.py

@@ -2,7 +2,7 @@ import streamlit as st
 from PIL import Image
 import numpy as np
 import torch
-
+from sklearn.utils.extmath import softmax
 import open_clip
 
 #from transformers import CLIPProcessor, CLIPModel
@@ -54,7 +54,7 @@ def knn_get_score(knn, k, cat, vec):
     cos_sim_table = vec @ allvecs.T
     if debug: st.write('cos_sim_table.shape', cos_sim_table.shape)
     # Get sorted array indices by similiarity in descending order
-    sortinds = np.flip(np.argsort(cos_sim_table))
+    sortinds = np.flip(np.argsort(cos_sim_table, axis=1), axis=1)
     if debug: st.write('sortinds.shape', sortinds.shape)
     # Get corresponding scores for the sorted vectors
     kscores = scores[sortinds][:k]