Update app.py
Browse files
app.py
CHANGED
@@ -44,11 +44,10 @@ class T5Response(BaseModel):
|
|
44 |
@app.post("/modify_query", response_model=ModifyQueryResponse)
|
45 |
async def modify_query(request: ModifyQueryRequest):
|
46 |
try:
|
47 |
-
|
48 |
-
embeddings
|
49 |
-
return ModifyQueryResponse(embeddings=[emb.tolist() for emb in embeddings])
|
50 |
except Exception as e:
|
51 |
-
raise HTTPException(status_code=500, detail=
|
52 |
|
53 |
@app.post("/modify_query_v3", response_model=ModifyQueryResponse)
|
54 |
async def modify_query_v3(request: ModifyQueryRequest_v3):
|
|
|
44 |
@app.post("/modify_query", response_model=ModifyQueryResponse)
|
45 |
async def modify_query(request: ModifyQueryRequest):
|
46 |
try:
|
47 |
+
binary_embeddings = model.encode([request.query_string], precision="binary")
|
48 |
+
return ModifyQueryResponse(embeddings=binary_embeddings[0].tolist())
|
|
|
49 |
except Exception as e:
|
50 |
+
raise HTTPException(status_code=500, detail=str(e))
|
51 |
|
52 |
@app.post("/modify_query_v3", response_model=ModifyQueryResponse)
|
53 |
async def modify_query_v3(request: ModifyQueryRequest_v3):
|