Spaces:

AWeirdDev
/

embeddings

Sleeping

AWeirdDev commited on Jul 19, 2024

Commit

5e25768

verified ·

1 Parent(s): eed902b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,7 +15,7 @@ class Req(BaseModel):
 @app.post("/embeddings")
 async def embeddings(req: Req):
     def do():
-        embedding_model.encode(req.input).tolist()
     return JSONResponse(
         await asyncio.to_thread(do)
     )

 @app.post("/embeddings")
 async def embeddings(req: Req):
     def do():
+        return embedding_model.encode(req.input).tolist()
     return JSONResponse(
         await asyncio.to_thread(do)
     )