ramonpzg commited on
Commit
416ff82
1 Parent(s): c77f36f

changes numpy vectr to python list

Browse files
Files changed (1) hide show
  1. app.py +1 -2
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