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):
|