GitHub Actions commited on
Commit
99da483
·
1 Parent(s): cc940a3

🚀 Deploy embedder from GitHub Actions - 2025-10-29 01:20:01

Browse files
Files changed (1) hide show
  1. main.py +3 -3
main.py CHANGED
@@ -300,7 +300,7 @@ async def generate_embedding(file: UploadFile = File(...)):
300
 
301
  return EmbeddingResponse(
302
  embedding=embedding.tolist(),
303
- model=MODEL_NAME,
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=MODEL_NAME
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=MODEL_NAME,
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
  )