mdanish commited on
Commit
cfef3c1
·
verified ·
1 Parent(s): b26dae6

Upload app.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -113,6 +113,7 @@ def main():
113
  # Normalize vector
114
  vec /= vec.norm(dim=-1, keepdim=True)
115
  st.write(vec.shape)
 
116
  k = 40
117
  for cat in ['walkability']:
118
  st.write(cat, 'rating =', knn_get_score(knn, k, cat, vec))
 
113
  # Normalize vector
114
  vec /= vec.norm(dim=-1, keepdim=True)
115
  st.write(vec.shape)
116
+ vec = vec.numpy()
117
  k = 40
118
  for cat in ['walkability']:
119
  st.write(cat, 'rating =', knn_get_score(knn, k, cat, vec))