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))
|