berkaygkv54 commited on
Commit
eb4c8c4
1 Parent(s): 9a08670

streamlit component changes

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -37,5 +37,5 @@ if is_clicked:
37
  with torch.no_grad():
38
  ranking = torch.tensor(audio_vectors) @ torch.tensor(text_embed).t()
39
  ranking = ranking[:, 0].reshape(-1, 1)
40
- dataframe = pd.DataFrame(ranking, columns=[session.text_input], index=song_names).nlargest(int(session.slider_count), session.text_input).drop(columns=session.text_input)
41
  st.dataframe(dataframe)
 
37
  with torch.no_grad():
38
  ranking = torch.tensor(audio_vectors) @ torch.tensor(text_embed).t()
39
  ranking = ranking[:, 0].reshape(-1, 1)
40
+ dataframe = pd.DataFrame(ranking, columns=[session.text_input], index=song_names).nlargest(int(session.slider_count), session.text_input).rename(columns={session.text_input: "score"})
41
  st.dataframe(dataframe)