Update app.py

app.py

@@ -3,7 +3,7 @@ import pandas as pd
 import matplotlib.pyplot as plt
 
 # Load the data from the CSV file
-@st.cache
+@st.cache_data
 def load_data():
     df = pd.read_csv("llm_data.csv")  # Update with your CSV file path
     return df
@@ -27,7 +27,7 @@ df['Example cost'] = df.apply(lambda row: calculate_example_cost(input_text, out
 # Display sorted LLM costs
 st.write("Sorted LLM Costs:")
 sorted_df = df.sort_values(by='Example cost', ascending=False)
-st.write(sorted_df[['Company', 'Model', 'Example cost']])
+st.write(sorted_df[['Company', 'Model', 'Example cost ($USD)']])
 
 # Plot visualization
 st.write("Visualization of LLM Costs:")