Update token_counter.py
Browse files- 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 = "
|
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"]
|