TinySuitStarfish commited on
Commit
f11f246
1 Parent(s): eefd3ca

Resolving more issues

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -50,7 +50,7 @@ if st.button("GO!"):
50
  prompt = process_text(topic)
51
  model = get_model()
52
  prompt_embedded = model.encode(prompt)
53
- df_embed = pd.read_pickle('../data/embeddings_pkl.pkl').drop_duplicates(subset=['titles'])
54
  df_embed["similarity_scores"] = df_embed["abstracts_embeddings"].apply(lambda x: get_cosine_similarity(x, prompt_embedded))
55
  top_n = df_embed.nlargest(number, 'similarity_scores').head(5)["titles"].to_list()
56
  st.text(" ")
 
50
  prompt = process_text(topic)
51
  model = get_model()
52
  prompt_embedded = model.encode(prompt)
53
+ df_embed = pd.read_pickle('data/embeddings_pkl.pkl').drop_duplicates(subset=['titles'])
54
  df_embed["similarity_scores"] = df_embed["abstracts_embeddings"].apply(lambda x: get_cosine_similarity(x, prompt_embedded))
55
  top_n = df_embed.nlargest(number, 'similarity_scores').head(5)["titles"].to_list()
56
  st.text(" ")