Update app.py
Browse files
app.py
CHANGED
@@ -1,10 +1,13 @@
|
|
1 |
from fastapi import FastAPI, Request, Query
|
2 |
from fastapi.templating import Jinja2Templates
|
3 |
from fastapi import File, UploadFile
|
|
|
|
|
4 |
from pydantic import BaseModel
|
5 |
from sentence_transformers import SentenceTransformer
|
6 |
import faiss
|
7 |
import numpy as np
|
|
|
8 |
|
9 |
app = FastAPI()
|
10 |
model = SentenceTransformer('paraphrase-MiniLM-L6-v2')
|
@@ -40,11 +43,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 |
-
|
44 |
return {
|
45 |
"distances": distances[0].tolist(),
|
46 |
"indices": indices[0].tolist(),
|
47 |
-
|
48 |
}
|
49 |
|
50 |
#########################
|
|
|
1 |
from fastapi import FastAPI, Request, Query
|
2 |
from fastapi.templating import Jinja2Templates
|
3 |
from fastapi import File, UploadFile
|
4 |
+
from fastapi.responses import FileResponse
|
5 |
+
|
6 |
from pydantic import BaseModel
|
7 |
from sentence_transformers import SentenceTransformer
|
8 |
import faiss
|
9 |
import numpy as np
|
10 |
+
import json
|
11 |
|
12 |
app = FastAPI()
|
13 |
model = SentenceTransformer('paraphrase-MiniLM-L6-v2')
|
|
|
43 |
def search_string(request: SearchRequest):
|
44 |
embedding = model.encode([request.text])
|
45 |
distances, indices = index.search(np.array(embedding), request.n)
|
46 |
+
found_documents = index.reconstruct_n(indices[0], request.n)
|
47 |
return {
|
48 |
"distances": distances[0].tolist(),
|
49 |
"indices": indices[0].tolist(),
|
50 |
+
"documents": found_documents.tolist()
|
51 |
}
|
52 |
|
53 |
#########################
|