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