Spaces:
Runtime error
Runtime error
TinySuitStarfish
commited on
Commit
•
f11f246
1
Parent(s):
eefd3ca
Resolving more issues
Browse files
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('
|
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(" ")
|