Chris4K commited on
Commit
a77a127
1 Parent(s): 1894ae6

Update token_counter.py

Browse files
Files changed (1) hide show
  1. token_counter.py +1 -1
token_counter.py CHANGED
@@ -2,7 +2,7 @@ from transformers import AutoTokenizer
2
  from transformers import Tool
3
 
4
  class TokenCounterTool(Tool):
5
- name = "text_generator"
6
  description = "This is a tool for counting token used by a prompt. It takes a prompt as input and returns the generated text."
7
  inputs = ["text"]
8
  outputs = ["text"]
 
2
  from transformers import Tool
3
 
4
  class TokenCounterTool(Tool):
5
+ name = "token_counter"
6
  description = "This is a tool for counting token used by a prompt. It takes a prompt as input and returns the generated text."
7
  inputs = ["text"]
8
  outputs = ["text"]