# count_tokens.py
import sys
import tiktoken

def count_tokens(text: str) -> int:
    encoding = tiktoken.encoding_for_model("gpt-4")
    return len(encoding.encode(text))

if __name__ == "__main__":
    input_text = sys.argv[1]
    print(count_tokens(input_text))