added two more envs
Browse files
app.py
CHANGED
@@ -7,7 +7,7 @@ from huggingface_hub import HfApi, hf_hub_download
|
|
7 |
from huggingface_hub.repocard import metadata_load
|
8 |
|
9 |
|
10 |
-
RL_ENVS = ['LunarLander-v2','CarRacing-v0','MountainCar-v0']
|
11 |
|
12 |
with open('app.css','r') as f:
|
13 |
BLOCK_CSS = f.read()
|
@@ -18,6 +18,9 @@ LOADED_MODEL_IDS = {rl_env:[] for rl_env in RL_ENVS}
|
|
18 |
# Based on Omar Sanseviero work
|
19 |
# Make model clickable link
|
20 |
def make_clickable_model(model_name):
|
|
|
|
|
|
|
21 |
link = "https://huggingface.co/" + model_name
|
22 |
return f'<a target="_blank" href="{link}">{model_name}</a>'
|
23 |
|
@@ -201,7 +204,9 @@ def reload_leaderboard(rl_env):
|
|
201 |
|
202 |
RL_DETAILS ={'CarRacing-v0':{'title':" The Car Racing ποΈ Leaderboard π",'data':get_data_per_env('CarRacing-v0')},
|
203 |
'MountainCar-v0':{'title':"The Mountain Car β°οΈ π Leaderboard π",'data':get_data_per_env('MountainCar-v0')},
|
204 |
-
'LunarLander-v2':{'title':"
|
|
|
|
|
205 |
}
|
206 |
|
207 |
|
|
|
7 |
from huggingface_hub.repocard import metadata_load
|
8 |
|
9 |
|
10 |
+
RL_ENVS = ['LunarLander-v2','CarRacing-v0','MountainCar-v0', 'BipedalWalker-v3','FrozenLake-v1']
|
11 |
|
12 |
with open('app.css','r') as f:
|
13 |
BLOCK_CSS = f.read()
|
|
|
18 |
# Based on Omar Sanseviero work
|
19 |
# Make model clickable link
|
20 |
def make_clickable_model(model_name):
|
21 |
+
# remove user from model name
|
22 |
+
model_name = ' '.join(model_name.split('/')[1:])
|
23 |
+
|
24 |
link = "https://huggingface.co/" + model_name
|
25 |
return f'<a target="_blank" href="{link}">{model_name}</a>'
|
26 |
|
|
|
204 |
|
205 |
RL_DETAILS ={'CarRacing-v0':{'title':" The Car Racing ποΈ Leaderboard π",'data':get_data_per_env('CarRacing-v0')},
|
206 |
'MountainCar-v0':{'title':"The Mountain Car β°οΈ π Leaderboard π",'data':get_data_per_env('MountainCar-v0')},
|
207 |
+
'LunarLander-v2':{'title':"The Lunar Lander π Leaderboard π",'data':get_data_per_env('LunarLander-v2')},
|
208 |
+
'BipedalWalker-v3':{'title':"The BipedalWalker Leaderboard π",'data':get_data_per_env('BipedalWalker-v3')},
|
209 |
+
'FrozenLake-v1':{'title':"The FrozenLake Leaderboard π",'data':get_data_per_env('FrozenLake-v1')}
|
210 |
}
|
211 |
|
212 |
|