merve HF staff commited on
Commit
01aa20e
1 Parent(s): d0eb8f8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +16 -20
app.py CHANGED
@@ -58,27 +58,23 @@ corpus_embeddings.shape
58
  def find_sentences(query):
59
  query_embedding = model.encode(query)
60
  hits = util.semantic_search(query_embedding, corpus_embeddings, top_k=5)
61
- hits = hits[0]
62
-
63
- for hit in hits:
64
- corpus_id = hit['corpus_id']
65
-
66
- # Find source document based on sentence index
67
- count = 0
68
- for idx, c in enumerate(sentence_count):
69
- count+=c
70
- if (corpus_id > count-1):
71
- continue
72
- else:
73
- doc = corpus[idx]
74
- #return doc
75
-
76
- break
77
- return corpus[corpus_id]
78
-
79
-
80
-
81
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
82
 
83
 
84
 
 
58
  def find_sentences(query):
59
  query_embedding = model.encode(query)
60
  hits = util.semantic_search(query_embedding, corpus_embeddings, top_k=5)
61
+ hits = hits[0][0]
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
62
 
63
+
64
+ corpus_id = hit['corpus_id']
65
+ # print(corpus[corpus_id], "(Score: {:.4f})".format(hit['score']))
66
+ # Find source document based on sentence index
67
+ count = 0
68
+ for idx, c in enumerate(sentence_count):
69
+ count+=c
70
+ if (corpus_id > count-1):
71
+ continue
72
+ else:
73
+ doc = corpus[idx]
74
+ return doc
75
+ #print(f"Document: {doc}, {count}")
76
+ break
77
+ return corpus[corpus_id]
78
 
79
 
80