Update app.py

app.py

@@ -4,36 +4,24 @@ import tiktoken
 encodings = tiktoken.list_encoding_names()
 encodings.reverse()
 
-def function(input, encoding):
-    tokenizer = tiktoken.get_encoding(encoding)
-    tokens = tokenizer.encode(input)
+
+def calc(input: str, encoding: str) -> int:
+    tokens = tiktoken.get_encoding(encoding).encode(input)
     return len(tokens)
-    
-value1 = gr.Textbox(lines=6, label="Input", placeholder="Input the text...")
-value2 = gr.Dropdown(
-    label="Encoding",
-    choices=encodings,
-    default="cl100k_base",
-    description="The encoding to use. (GPT-3.5 and GPT-4 use cl100k_base as their encoding.)"
-)
-value3 = gr.Textbox(lines=3, label="Output", placeholder="Output...")
-           
-examples = [
-    ["The only way to do great work is to love what you do. - Steve Jobs"],
-    ["In the end, we will remember not the words of our enemies, but the silence of our friends. - Martin Luther King Jr."],
-    ["Success is not final, failure is not fatal: It is the courage to continue that counts. - Winston Churchill"],
-    ["The greatest glory in living lies not in never falling, but in rising every time we fall. - Nelson Mandela"],
-    ["The best and most beautiful things in the world cannot be seen or even touched - they must be felt with the heart. - Helen Keller"]
-]
 
-demo = gr.Interface(
-    fn=function,
-    inputs=[value1, value2],
-    outputs=value3,
-    title="ChatGPT Token Calculator",
-    examples=examples,
-    description="Calculate the number of tokens in a text string.o",
-    allow_flagging="never"
-)
 
-demo.launch(debug=True)
+gr.Interface(
+    calc,
+    inputs=[
+        gr.Text(label="Input", lines=6, placeholder="Enter text here..."),
+        gr.Dropdown(
+            label="Encoding",
+            choices=encodings,
+            value="cl100k_base",
+            info="The encoding to use. (GPT-3.5 and GPT-4 use cl100k_base as their encoding.)",
+        ),
+    ],
+    outputs=gr.Number(label="Output"),
+    title="Tiktoken Calculator",
+    allow_flagging="never",
+).launch()