Spaces:
Sleeping
Sleeping
changes numpy vectr to python list
Browse files
app.py
CHANGED
@@ -70,11 +70,10 @@ if st.button("Generate Some Music!"):
|
|
70 |
with torch.no_grad():
|
71 |
vectr = classifier.model(**features, output_hidden_states=True).hidden_states[-1].mean(dim=1)[0]
|
72 |
|
73 |
-
# vectr = embedding[0]
|
74 |
|
75 |
results = client.search(
|
76 |
collection_name="music_vectors",
|
77 |
-
query_vector=vectr,
|
78 |
limit=10
|
79 |
)
|
80 |
|
|
|
70 |
with torch.no_grad():
|
71 |
vectr = classifier.model(**features, output_hidden_states=True).hidden_states[-1].mean(dim=1)[0]
|
72 |
|
|
|
73 |
|
74 |
results = client.search(
|
75 |
collection_name="music_vectors",
|
76 |
+
query_vector=vectr.tolist(),
|
77 |
limit=10
|
78 |
)
|
79 |
|