Spaces:
Restarting
on
L4
Restarting
on
L4
Update app.py
#3
by
Amlan99
- opened
app.py
CHANGED
@@ -85,7 +85,7 @@ def embed(req: EmbedRequest):
|
|
85 |
truncate_dim=req.truncate_dim,
|
86 |
)
|
87 |
# outputs[0] = (num_vectors, hidden_dim)
|
88 |
-
pooled = outputs[0].mean(dim=0).cpu()
|
89 |
return {"embeddings": [pooled]}
|
90 |
|
91 |
# Case 2: Passage → sliding window, token-level embeddings
|
|
|
85 |
truncate_dim=req.truncate_dim,
|
86 |
)
|
87 |
# outputs[0] = (num_vectors, hidden_dim)
|
88 |
+
pooled = outputs[0].mean(dim=0).cpu()
|
89 |
return {"embeddings": [pooled]}
|
90 |
|
91 |
# Case 2: Passage → sliding window, token-level embeddings
|