Update app.py

updating from json to pkl

app.py

@@ -18,22 +18,8 @@ import pandas as pd
 import plotly.express as px
 import numpy as np
 
-with open('carbon-models-2.json', 'r') as fp:
-    carbondict= json.load(fp)
-
-models= [k.split('/')[1] for k in carbondict.keys()]
-carbon = [v[1] for v in carbondict.values()]
-carbon = [4009.5 if x== '4' else x for x in carbon]
-tasks = [v[0] for v in carbondict.values()]
-tasks = ["other" if len(x) < 2 else x for x in tasks]
-
-carbon_df = pd.DataFrame(
-    {'name': models,
-     'task': tasks,
-     'carbon': carbon
-    })
-
-carbon_df.drop(carbon_df.loc[carbon_df['task']=='other'].index, inplace=True)
+carbon_df= pd.read_pickle('./data/carbon_df.pkl')
+carbon_df.drop(carbon_df.loc[carbon_df['task']==''].index, inplace=True)
 
 st.set_page_config(
     page_title="Comparing the Carbon Footprint of Transformers",
@@ -43,16 +29,22 @@ st.set_page_config(
 
 st.title("Hugging Face Carbon Compare Tool")
 
-
 # Get the sidebar details
 with st.sidebar.expander("Models", expanded=True):
     st.image('./hf-earth.png')
     # choose a dataset to analyze
+    models= [m.split('/')[1].replace('autonlp-','') for m in carbon_df['name']]
     model_name = st.selectbox(
         f"Choose model to explore:",
         models)
 
+with st.expander("Model Comparison", expanded=False):
+
+    st.markdown("### Here is how the model " + model_name + " compares to other models:")
+    fig_model = px.bar(carbon_df.sort_values(by=['carbon']), x=models, y='carbon', hover_name= models, color_discrete_map = {model_name : 'red'})
+    st.plotly_chart(fig_model, use_container_width=True)
 
-fig = px.box(carbon_df, x=carbon_df['task'], y=carbon_df['carbon'], color='task', hover_name=carbon_df['name'])
-#fig.update_traces(quartilemethod="exclusive") # or "inclusive", or "linear" by default
-st.plotly_chart(fig, use_container_width=True)
+with st.expander("Task Comparison", expanded=False):
+    fig = px.box(carbon_df, x=carbon_df['task'], y=carbon_df['carbon'], color='task', hover_name=carbon_df['name'])
+    #fig.update_traces(quartilemethod="exclusive") # or "inclusive", or "linear" by default
+    st.plotly_chart(fig, use_container_width=True)