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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
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
- #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
  #########################
 
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
  #########################