Tonyivan commited on
Commit
f57466f
·
verified ·
1 Parent(s): 7c6c308

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
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
- # Generate embeddings
48
- embeddings = model.encode([request.query_string])
49
- return ModifyQueryResponse(embeddings=[emb.tolist() for emb in embeddings])
50
  except Exception as e:
51
- raise HTTPException(status_code=500, detail=f"Error in modifying query: {str(e)}")
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):