Update app.py

app.py

@@ -74,7 +74,7 @@ def embed(req: EmbedRequest):
                 prompt_name=req.prompt_name or "query",
                 return_multivector=False
             )
-        return {"embeddings": emb.tolist()}  # shape: (1, hidden_dim)
+        return {"embeddings": emb}  # shape: (1, hidden_dim)
 
     # -----------------------------
     # Case 2: Long passages → sliding window token embeddings
@@ -112,7 +112,7 @@ def embed(req: EmbedRequest):
         position += max_len - stride
 
     full_embeddings = torch.cat(embeddings, dim=0)  # (total_tokens, hidden_dim)
-    return {"embeddings": full_embeddings.tolist()}
+    return {"embeddings": full_embeddings}
 
 
 # -----------------------------