wjm55 commited on
Commit
12288ff
·
1 Parent(s): 7aee21c

Update embed_text function to return a list from model encoding

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,5 +11,5 @@ class EmbedRequest(BaseModel):
11
 
12
  @app.post("/vectorize")
13
  async def embed_text(request: EmbedRequest):
14
- return model.encode(request.text)
15
 
 
11
 
12
  @app.post("/vectorize")
13
  async def embed_text(request: EmbedRequest):
14
+ return model.encode(request.text).tolist()
15