Spaces:
Running
Running
fahmiaziz98
commited on
Commit
·
9d130f5
1
Parent(s):
af36df4
fix typo
Browse files
app.py
CHANGED
|
@@ -200,7 +200,7 @@ async def create_embedding(request: EmbedRequest):
|
|
| 200 |
)
|
| 201 |
|
| 202 |
# Dense embedding
|
| 203 |
-
embedding = model.embed_documents(
|
| 204 |
processing_time = time.time() - start_time
|
| 205 |
|
| 206 |
return EmbedResponse(
|
|
@@ -317,4 +317,4 @@ async def root():
|
|
| 317 |
return {
|
| 318 |
"message": "Unified Embedding API - Dense & Sparse Embeddings",
|
| 319 |
"version": "3.0.0"
|
| 320 |
-
}
|
|
|
|
| 200 |
)
|
| 201 |
|
| 202 |
# Dense embedding
|
| 203 |
+
embedding = model.embed_documents(texts=[request.text], prompt=request.prompt)[0]
|
| 204 |
processing_time = time.time() - start_time
|
| 205 |
|
| 206 |
return EmbedResponse(
|
|
|
|
| 317 |
return {
|
| 318 |
"message": "Unified Embedding API - Dense & Sparse Embeddings",
|
| 319 |
"version": "3.0.0"
|
| 320 |
+
}
|