Spaces:
Sleeping
Sleeping
GitHub Actions
commited on
Commit
·
99da483
1
Parent(s):
cc940a3
🚀 Deploy embedder from GitHub Actions - 2025-10-29 01:20:01
Browse files
main.py
CHANGED
|
@@ -300,7 +300,7 @@ async def generate_embedding(file: UploadFile = File(...)):
|
|
| 300 |
|
| 301 |
return EmbeddingResponse(
|
| 302 |
embedding=embedding.tolist(),
|
| 303 |
-
model=
|
| 304 |
inference_time_ms=round(inference_time, 2)
|
| 305 |
)
|
| 306 |
|
|
@@ -375,7 +375,7 @@ async def generate_batch_embeddings(files: List[UploadFile] = File(...)):
|
|
| 375 |
embeddings=embeddings,
|
| 376 |
count=len(embeddings),
|
| 377 |
total_time_ms=round(total_time, 2),
|
| 378 |
-
model=
|
| 379 |
)
|
| 380 |
|
| 381 |
except HTTPException:
|
|
@@ -430,7 +430,7 @@ async def generate_text_embedding(request: TextEmbeddingRequest):
|
|
| 430 |
|
| 431 |
return TextEmbeddingResponse(
|
| 432 |
embedding=embedding.tolist(),
|
| 433 |
-
model=
|
| 434 |
inference_time_ms=round(inference_time, 2),
|
| 435 |
text=request.text
|
| 436 |
)
|
|
|
|
| 300 |
|
| 301 |
return EmbeddingResponse(
|
| 302 |
embedding=embedding.tolist(),
|
| 303 |
+
model=model_module.model_metadata["model_name"],
|
| 304 |
inference_time_ms=round(inference_time, 2)
|
| 305 |
)
|
| 306 |
|
|
|
|
| 375 |
embeddings=embeddings,
|
| 376 |
count=len(embeddings),
|
| 377 |
total_time_ms=round(total_time, 2),
|
| 378 |
+
model=model_module.model_metadata["model_name"]
|
| 379 |
)
|
| 380 |
|
| 381 |
except HTTPException:
|
|
|
|
| 430 |
|
| 431 |
return TextEmbeddingResponse(
|
| 432 |
embedding=embedding.tolist(),
|
| 433 |
+
model=model_module.model_metadata["model_name"],
|
| 434 |
inference_time_ms=round(inference_time, 2),
|
| 435 |
text=request.text
|
| 436 |
)
|