Elvan Selvano commited on
Commit
502d5c9
β€’
1 Parent(s): c755035

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
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.sort_values(by='score', ascending=False, inplace=True)
 
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)