Spaces:
Sleeping
Sleeping
wjm55
commited on
Commit
·
12288ff
1
Parent(s):
7aee21c
Update embed_text function to return a list from model encoding
Browse files
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 |
|