Remove tokencost

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
-from tokencost import count_message_tokens, count_string_tokens, calculate_prompt_cost, calculate_completion_cost
+import pandas as pd
 
+model_data = pd.read_json("model_prices.json")
 
 def compute_all(prompt_string, completion_string, model):
     prompt_cost = calculate_prompt_cost(prompt_string, model)
@@ -13,9 +14,12 @@ def compute_all(prompt_string, completion_string, model):
 
 
 with gr.Blocks(theme='soft') as demo:
+    gr.Markdown("""### Text-to-$: calculate the price of your LLM runs
+Based on data from the great [tokencost](https://github.com/AgentOps-AI/tokencost/blob/main/tokencost/model_prices.json).
+""")
     with gr.Row():
         with gr.Column():
-            prompt = gr.Textbox(value="Hello world")
+            prompt = gr.Textbox(value="Lorem ipsum dolor sit amet...")
             completion = gr.Textbox(value="...")
 
             model = gr.Dropdown(value="gpt-3.5-turbo", choices=["gpt-3.5-turbo", "gpt-3.5-turbo"])

requirements.txt

@@ -1 +1 @@
-tokencost
+pandas