Spaces:
Runtime error
Runtime error
Elvan Selvano
commited on
Commit
·
502d5c9
1
Parent(s):
c755035
Update app.py
Browse files
app.py
CHANGED
@@ -24,10 +24,11 @@ def semantic_search(model, sentence, corpus_embeddings):
|
|
24 |
return pd.DataFrame(hits[0])
|
25 |
|
26 |
def get_similarity_score(model, data, query, corpus_embeddings):
|
27 |
-
"""Get similarity score for each data point"""
|
28 |
hits = semantic_search(model, [query], corpus_embeddings)
|
29 |
result = pd.merge(data, hits, left_on='ID', right_on='corpus_id')
|
30 |
-
result.
|
|
|
31 |
return result
|
32 |
|
33 |
@st.cache(allow_output_mutation=True)
|
|
|
24 |
return pd.DataFrame(hits[0])
|
25 |
|
26 |
def get_similarity_score(model, data, query, corpus_embeddings):
|
27 |
+
"""Get similarity score for each data point and sort by similarity score and day"""
|
28 |
hits = semantic_search(model, [query], corpus_embeddings)
|
29 |
result = pd.merge(data, hits, left_on='ID', right_on='corpus_id')
|
30 |
+
result['Last Day'] = pd.to_datetime(result['Last Day'], format='%d/%m/%Y')
|
31 |
+
result.sort_values(by=['score', 'Last Day'], ascending=[False, True], inplace=True)
|
32 |
return result
|
33 |
|
34 |
@st.cache(allow_output_mutation=True)
|