Update token_counter.py

token_counter.py

@@ -10,6 +10,6 @@ class TokenCounterTool(Tool):
     def __call__(self, prompt: str):
        # token = os.environ['hf']
         tokenizer = AutoTokenizer.from_pretrained("lgaalves/gpt2-dolly")
-        tokens = tokenizer(input_text)["input_ids"]
+        tokens = tokenizer(prompt)["input_ids"]
         return f"{len(tokens)}"