chgrdj commited on
Commit
a466fad
·
verified ·
1 Parent(s): 399b79f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -25,6 +25,8 @@ top_k = st.number_input("Top K Results", min_value=1, max_value=len(corpus_domai
25
  # Perform Semantic Search
26
  if domain:
27
  query_emb = model.encode(domain)
 
 
28
  semantic_res = util.semantic_search(query_emb, corpus_embeddings, top_k=top_k)[0]
29
  ids = [r['corpus_id'] for r in semantic_res]
30
  scores = [r['score'] for r in semantic_res]
 
25
  # Perform Semantic Search
26
  if domain:
27
  query_emb = model.encode(domain)
28
+ print('QUERY',query_emb[:5])
29
+ print('CORPUS',corpus_embeddings[:1,:5])
30
  semantic_res = util.semantic_search(query_emb, corpus_embeddings, top_k=top_k)[0]
31
  ids = [r['corpus_id'] for r in semantic_res]
32
  scores = [r['score'] for r in semantic_res]