Almaatla commited on
Commit
8fbb998
·
verified ·
1 Parent(s): a823e0a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -40,11 +40,11 @@ def embed_strings(request: EmbedRequest):
40
  def search_string(request: SearchRequest):
41
  embedding = model.encode([request.text])
42
  distances, indices = index.search(np.array(embedding), request.n)
43
- found_documents = index.reconstruct_n(indices[0], request.n)
44
  return {
45
  "distances": distances[0].tolist(),
46
  "indices": indices[0].tolist(),
47
- "documents": found_documents.tolist()
48
  }
49
 
50
  #########################
 
40
  def search_string(request: SearchRequest):
41
  embedding = model.encode([request.text])
42
  distances, indices = index.search(np.array(embedding), request.n)
43
+ #found_documents = index.reconstruct_n(indices[0], request.n)
44
  return {
45
  "distances": distances[0].tolist(),
46
  "indices": indices[0].tolist(),
47
+ #"documents": found_documents.tolist()
48
  }
49
 
50
  #########################