sasha HF Staff commited on
Commit
4a4c1b5
·
verified ·
1 Parent(s): aaf2e79

Update app.py

Browse files

reducing columns

Files changed (1) hide show
  1. app.py +7 -6
app.py CHANGED
@@ -94,7 +94,7 @@ def print_existing_models():
94
  requests= load_dataset("EnergyStarAI/requests_debug", split="test", token=TOKEN)
95
  requests_dset = requests.to_pandas()
96
  model_list= requests_dset[requests_dset['status'] == 'COMPLETED']
97
- return model_list
98
 
99
  def get_leaderboard_models():
100
  path = r'leaderboard_v0_data/energy'
@@ -103,7 +103,7 @@ def get_leaderboard_models():
103
  for filename in filenames:
104
  data.append(pd.read_csv(filename))
105
  leaderboard_data = pd.concat(data, ignore_index=True)
106
- return leaderboard_data
107
 
108
 
109
  with gr.Blocks() as demo:
@@ -136,8 +136,9 @@ with gr.Blocks() as demo:
136
  outputs=submission_result,
137
  )
138
  with gr.Row():
139
- with gr.Accordion("Models that are in the latest leaderboard version:", open = False):
140
- gr.Dataframe(get_leaderboard_models())
141
- with gr.Accordion("Models that have been benchmarked lately:", open = False):
142
- gr.Dataframe(print_existing_models())
 
143
  demo.launch()
 
94
  requests= load_dataset("EnergyStarAI/requests_debug", split="test", token=TOKEN)
95
  requests_dset = requests.to_pandas()
96
  model_list= requests_dset[requests_dset['status'] == 'COMPLETED']
97
+ return model_list[['model','task']]
98
 
99
  def get_leaderboard_models():
100
  path = r'leaderboard_v0_data/energy'
 
103
  for filename in filenames:
104
  data.append(pd.read_csv(filename))
105
  leaderboard_data = pd.concat(data, ignore_index=True)
106
+ return leaderboard_data[['model','task']]
107
 
108
 
109
  with gr.Blocks() as demo:
 
136
  outputs=submission_result,
137
  )
138
  with gr.Row():
139
+ with gr.Column():
140
+ with gr.Accordion("Models that are in the latest leaderboard version:", open = False):
141
+ gr.Dataframe(get_leaderboard_models())
142
+ with gr.Accordion("Models that have been benchmarked lately:", open = False):
143
+ gr.Dataframe(print_existing_models())
144
  demo.launch()