Update app.py
Browse files
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]
|