Quentin Gallouédec commited on
Commit
041b899
1 Parent(s): 9beca9b
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -17,8 +17,6 @@ logger = setup_logger(__name__)
17
 
18
  API = HfApi(token=os.environ.get("TOKEN"))
19
  RESULTS_REPO = f"open-rl-leaderboard/results"
20
-
21
-
22
  ALL_ENV_IDS = {
23
  "Atari": [
24
  "BeamRiderNoFrameskip-v4",
@@ -91,11 +89,12 @@ def select_env(df: pd.DataFrame, env_id: str):
91
  df.loc[index, "user_id"] = f"[{user_id}](https://huggingface.co/{user_id})"
92
  df.loc[index, "model_id"] = f"[{model_id}](https://huggingface.co/{user_id}/{model_id})"
93
 
 
94
  df = df[["ranking", "user_id", "model_id", "mean_episodic_return"]]
95
  return df.values.tolist()
96
 
97
 
98
- with gr.Blocks(js=dark_mode_gradio_js) as demo:
99
  gr.HTML(TITLE)
100
  gr.Markdown(INTRODUCTION_TEXT, elem_classes="markdown-text")
101
  with gr.Tabs(elem_classes="tab-buttons") as tabs:
 
17
 
18
  API = HfApi(token=os.environ.get("TOKEN"))
19
  RESULTS_REPO = f"open-rl-leaderboard/results"
 
 
20
  ALL_ENV_IDS = {
21
  "Atari": [
22
  "BeamRiderNoFrameskip-v4",
 
89
  df.loc[index, "user_id"] = f"[{user_id}](https://huggingface.co/{user_id})"
90
  df.loc[index, "model_id"] = f"[{model_id}](https://huggingface.co/{user_id}/{model_id})"
91
 
92
+ # Keep only the relevant columns
93
  df = df[["ranking", "user_id", "model_id", "mean_episodic_return"]]
94
  return df.values.tolist()
95
 
96
 
97
+ with gr.Blocks() as demo:
98
  gr.HTML(TITLE)
99
  gr.Markdown(INTRODUCTION_TEXT, elem_classes="markdown-text")
100
  with gr.Tabs(elem_classes="tab-buttons") as tabs: