sasha HF staff commited on
Commit
7e00f76
β€’
1 Parent(s): ca42da8

ok let's do like this

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -23,8 +23,8 @@ def get_plots(task):
23
  def get_model_names(task_data):
24
  #TODO: add link to results in model card of each model
25
  task_df= pd.read_csv('data/energy/'+task_data)
26
- model_names = task_df['model'].tolist()
27
- model_names = [list(set(model_names))]
28
  return model_names
29
 
30
 
 
23
  def get_model_names(task_data):
24
  #TODO: add link to results in model card of each model
25
  task_df= pd.read_csv('data/energy/'+task_data)
26
+ task_df=task_df.drop_duplicates(subset=['model'])
27
+ model_names = task_df[['model']]
28
  return model_names
29
 
30