bgamazay commited on
Commit
2be0753
·
verified ·
1 Parent(s): f761d9d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -4
app.py CHANGED
@@ -39,8 +39,6 @@ def make_link(mname):
39
 
40
  def get_plots(task):
41
  df = pd.read_csv('data/energy/' + task)
42
- if df.columns[0].startswith("Unnamed:"):
43
- df = df.iloc[:, 1:]
44
  df['energy_score'] = df['energy_score'].astype(int)
45
  df['Display Model'] = df['model'].apply(lambda m: m.split('/')[-1])
46
 
@@ -135,8 +133,6 @@ def get_all_model_names():
135
  all_df = pd.DataFrame()
136
  for task in tasks:
137
  df = pd.read_csv('data/energy/' + task)
138
- if df.columns[0].startswith("Unnamed:"):
139
- df = df.iloc[:, 1:]
140
  df['energy_score'] = df['energy_score'].astype(int)
141
  df['GPU Energy (Wh)'] = df['total_gpu_energy'].apply(lambda x: f"{x:.4f}")
142
  df['Model'] = df['model'].apply(make_link)
 
39
 
40
  def get_plots(task):
41
  df = pd.read_csv('data/energy/' + task)
 
 
42
  df['energy_score'] = df['energy_score'].astype(int)
43
  df['Display Model'] = df['model'].apply(lambda m: m.split('/')[-1])
44
 
 
133
  all_df = pd.DataFrame()
134
  for task in tasks:
135
  df = pd.read_csv('data/energy/' + task)
 
 
136
  df['energy_score'] = df['energy_score'].astype(int)
137
  df['GPU Energy (Wh)'] = df['total_gpu_energy'].apply(lambda x: f"{x:.4f}")
138
  df['Model'] = df['model'].apply(make_link)