sashavor commited on
Commit
bfdb601
1 Parent(s): 9f18eaf

updating logic

Browse files
Files changed (1) hide show
  1. app.py +7 -1
app.py CHANGED
@@ -32,8 +32,14 @@ st.title("Hugging Face Carbon Compare Tool")
32
  # Get the sidebar details
33
  with st.sidebar.expander("Models", expanded=True):
34
  st.image('./hf-earth.png')
 
35
  # choose a dataset to analyze
36
- models= [m.split('/')[1].replace('autonlp-','') for m in carbon_df['name']]
 
 
 
 
 
37
  model_name = st.selectbox(
38
  f"Choose model to explore:",
39
  models)
 
32
  # Get the sidebar details
33
  with st.sidebar.expander("Models", expanded=True):
34
  st.image('./hf-earth.png')
35
+ models=[]
36
  # choose a dataset to analyze
37
+ for m in carbon_df['name'].items():
38
+ try:
39
+ modelname= m.split('/')[1].replace('autonlp-','')
40
+ except:
41
+ modelname = m
42
+ models.append(modelname)
43
  model_name = st.selectbox(
44
  f"Choose model to explore:",
45
  models)