Spaces:
Sleeping
Sleeping
File size: 4,312 Bytes
e313dc5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 |
import gradio as gr
from benchmarks import benchmarks, update_environments
from model import submit_model
def clear_form():
return [None, None, None, None, None, None]
with gr.Blocks(theme=gr.themes.Soft()) as demo:
gr.Markdown("<center><h1 style='font-size: 40px;'>⚔️Reinforcement Learning Agent Arena⚔️</h1></center>")
gr.Markdown("""
# 📜📜📜 The workflow of RLArena
- Select a benchmark, an environment, and a specific version.
- Enter your github username and the link to your code.
- Click "Submit" to evaluate your agent.
# 🏆🏆🏆 Visit the leaderboard
- Accelerate your RL research with the well-organized benchmark scores.
# 💪💪💪 Submit your agent now!
""")
with gr.Row():
with gr.Column(scale=2):
with gr.Row():
github_username = gr.Textbox(label="Github Username", info="Please enter your github username, e.g., username.")
benchmark = gr.Dropdown(label="Benchmark", choices=list(benchmarks.keys()), info="Please select a benchmark, e.g., Procgen.")
with gr.Row():
environment = gr.Dropdown(label="Environment", choices=[], value=None, info="Please select an environment, e.g., Miner.")
version = gr.Dropdown(label="Version", choices=[], value=None, info="Please select a version, e.g., v0.")
with gr.Row():
training_steps = gr.Number(label="Training Steps", precision=0, info="Please enter the training steps, e.g., 1000000.")
code_link = gr.Textbox(label="Code Link", info="Example: https://github.com/username/repo, the link should be accessible.")
with gr.Row():
submit_button = gr.Button("Submit", variant="primary")
clear_button = gr.Button("Clear", variant="secondary")
with gr.Column(scale=1):
# file uploader
model_uploader = gr.File(label="Upload the agent here!")
output = gr.Textbox(label="Evaluation Result")
benchmark.change(
fn=update_environments,
inputs=benchmark,
outputs=[environment, version]
)
submit_button.click(submit_model, inputs=[github_username, benchmark, environment, version, training_steps, code_link, model_uploader], outputs=output)
clear_button.click(clear_form, inputs=[], outputs=[github_username, benchmark, environment, version, training_steps, code_link])
with gr.Row():
# add multiple images with html
html_images = """
<div><br><br></div>
# 🔥🔥🔥 Powered by
<div style="display: flex; flex-wrap: wrap; gap: 10px; justify-content: left;">
<div style="flex: 0 0 calc(25% - 10px); display: flex; justify-content: center; align-items: center;">
<img src="/file=static/logo_polyu.png" alt="Image 1" style="max-width: 100%; height: auto;">
</div>
<div style="flex: 0 0 calc(25% - 10px); display: flex; justify-content: center; align-items: center;">
<img src="/file=static/logo_sjtu.png" alt="Image 2" style="max-width: 100%; height: auto;">
</div>
<div style="flex: 0 0 calc(25% - 10px); display: flex; justify-content: center; align-items: center;">
<img src="/file=static/logo_eias.png" alt="Image 3" style="max-width: 100%; height: auto;">
</div>
<div style="flex: 0 0 calc(25% - 10px); display: flex; justify-content: center; align-items: center;">
<img src="/file=static/logo_idt.png" alt="Image 4" style="max-width: 100%; height: auto;">
</div>
<div style="flex: 0 0 calc(25% - 10px); display: flex; justify-content: center; align-items: center;">
<img src="/file=static/logo_ustc.png" alt="Image 5" style="max-width: 100%; height: auto;">
</div>
<div style="flex: 0 0 calc(25% - 10px); display: flex; justify-content: center; align-items: center;">
<img src="/file=static/logo_purdue.png" alt="Image 6" style="max-width: 100%; height: auto;">
</div>
</div>
"""
gr.Markdown(html_images)
demo.launch(allowed_paths=["./"]) |