amaye15 commited on
Commit
e8acd10
·
1 Parent(s): 1a276d9

Debug - Read Endpoint

Browse files
Files changed (1) hide show
  1. src/main.py +1 -1
src/main.py CHANGED
@@ -126,7 +126,7 @@ async def read_embeddings(
126
  """
127
  try:
128
  df = await huggingface_service.read_dataset(request.dataset_name)
129
- return df.to_dict(orient="records")
130
  except DatasetNotFoundError as e:
131
  logger.error(f"Dataset not found: {e}")
132
  raise HTTPException(status_code=404, detail=f"Dataset not found: {e}")
 
126
  """
127
  try:
128
  df = await huggingface_service.read_dataset(request.dataset_name)
129
+ return df.to_dict(orient="dict")
130
  except DatasetNotFoundError as e:
131
  logger.error(f"Dataset not found: {e}")
132
  raise HTTPException(status_code=404, detail=f"Dataset not found: {e}")