IshmamF commited on
Commit
806255e
1 Parent(s): ff9c342

Update song_matching.py

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