Spaces:
Runtime error
Runtime error
Quentin Gallouédec
commited on
Commit
·
041b899
1
Parent(s):
9beca9b
style
Browse files
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(
|
| 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:
|