bgamazay commited on
Commit
446b9c6
·
verified ·
1 Parent(s): 9046801

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -102,8 +102,8 @@ def main():
102
  df[['provider', 'model']] = df['model'].str.split(pat='/', n=1, expand=True)
103
  # Round total_gpu_energy to 3 decimal places and assign to 'energy'
104
  df['energy'] = df['total_gpu_energy'].round(3)
105
- # Use the energy_score column as 'score'
106
- df['score'] = df['energy_score'].astype(int)
107
  # Hardcode date and hardware
108
  df['date'] = "February 2025"
109
  df['hardware'] = "NVIDIA H100-80GB"
 
102
  df[['provider', 'model']] = df['model'].str.split(pat='/', n=1, expand=True)
103
  # Round total_gpu_energy to 3 decimal places and assign to 'energy'
104
  df['energy'] = df['total_gpu_energy'].round(3)
105
+ # Use the energy_score column as 'score' (fill missing values with 1 to avoid casting errors)
106
+ df['score'] = df['energy_score'].fillna(1).astype(int)
107
  # Hardcode date and hardware
108
  df['date'] = "February 2025"
109
  df['hardware'] = "NVIDIA H100-80GB"