Spaces:
Sleeping
Sleeping
Update song_matching.py
Browse files- song_matching.py +1 -1
song_matching.py
CHANGED
@@ -29,7 +29,7 @@ class SongMatcher:
|
|
29 |
# Shuffle the matched songs to get a random order
|
30 |
matched_songs = matched_songs.sample(frac=1).reset_index(drop=True)
|
31 |
|
32 |
-
matched_songs['similarity'] = matched_songs['seq'].apply(lambda x: util.pytorch_cos_sim(sim_model.encode(x), inputVector))
|
33 |
|
34 |
top_5 = matched_songs['similarity'].sort_values(ascending=False).head(5)
|
35 |
|
|
|
29 |
# Shuffle the matched songs to get a random order
|
30 |
matched_songs = matched_songs.sample(frac=1).reset_index(drop=True)
|
31 |
|
32 |
+
matched_songs['similarity'] = matched_songs['seq'].apply(lambda x: util.pytorch_cos_sim(self.sim_model.encode(x), inputVector))
|
33 |
|
34 |
top_5 = matched_songs['similarity'].sort_values(ascending=False).head(5)
|
35 |
|