moraxgiga commited on
Commit
e617887
1 Parent(s): ddf9938

Update TextGen/router.py

Browse files
Files changed (1) hide show
  1. TextGen/router.py +1 -1
TextGen/router.py CHANGED
@@ -35,5 +35,5 @@ def inference(item: Validation):
35
  return {
36
  "embeddings": embeddings,
37
  "time_taken": f"{time_taken:.2f} seconds",
38
- "Number_of_sentence_processed": len(x)
39
  }
 
35
  return {
36
  "embeddings": embeddings,
37
  "time_taken": f"{time_taken:.2f} seconds",
38
+ "Number_of_sentence_processed": len(item.prompt)
39
  }