Upload app.py with huggingface_hub

app.py

@@ -19,7 +19,7 @@ st.set_page_config(
     layout="wide"
 )
 
-debug = True
+debug = False
 
 #st.write("Available models:", open_clip.list_models())
 
@@ -60,6 +60,7 @@ def knn_get_score(knn, k, cat, vec):
     kscores = scores[sortinds][:k]
     if debug: st.write('kscores.shape', kscores.shape)
     # Get actual sorted similiarity scores
+    # (line copied from clip_retrieval_knn.py even though sortinds.shape[0] == 1 here)
     ksims = cos_sim_table[np.expand_dims(np.arange(sortinds.shape[0]), axis=1), sortinds]
     if debug: st.write('ksims.shape', ksims.shape)
     # Apply normalization after exponential formula
@@ -111,8 +112,8 @@ def main():
                 if debug: st.write(vec.shape)
                 vec = vec.numpy()
                 k = 40
-                for cat in ['walkability']:
-                    st.write(cat, 'rating =', knn_get_score(knn, k, cat, vec))
+                for cat in categories:
+                    st.write(cat, f'rating = {knn_get_score(knn, k, cat, vec):.1f}')
 
         except Exception as e:
             st.error(f"Error processing image: {str(e)}")