chrisjay commited on
Commit
391b960
1 Parent(s): e9f37ce

added two more envs

Browse files
Files changed (1) hide show
  1. app.py +7 -2
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':" The Lunar Lander 🌕 Leaderboard 🚀",'data':get_data_per_env('LunarLander-v2')}
 
 
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