sasha HF staff commited on
Commit
d01211c
β€’
1 Parent(s): 68c7721

adding parameter info

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -27,10 +27,12 @@ def make_link(mname):
27
 
28
  def get_model_names(task_data):
29
  #TODO: add link to results in model card of each model
30
- task_df= pd.read_csv('data/energy/'+task_data)
 
31
  task_df=task_df.drop_duplicates(subset=['model'])
 
32
  task_df['model'] = task_df['model'].apply(make_link)
33
- model_names = task_df[['model']]
34
  return model_names
35
 
36
  def format_params(num):
 
27
 
28
  def get_model_names(task_data):
29
  #TODO: add link to results in model card of each model
30
+ task_df= pd.read_csv('data/params/'+task_data)
31
+ task_df= task_df.rename(columns={"Link": "model"})
32
  task_df=task_df.drop_duplicates(subset=['model'])
33
+ task_df['parameters'] = task_df['parameters'].apply(format_params)
34
  task_df['model'] = task_df['model'].apply(make_link)
35
+ model_names = task_df[['model','parameters']]
36
  return model_names
37
 
38
  def format_params(num):