Upload app.py with huggingface_hub
Browse files
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))
|