mlabonne commited on
Commit
2b5e864
·
verified ·
1 Parent(s): 472bef9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -5
app.py CHANGED
@@ -370,15 +370,20 @@ GITHUB_TOKEN = os.environ["GITHUB_TOKEN"]
370
  RESULT_GIST_ID = "c491299e7b8a45a61ce5403a70cf8656"
371
  LEAD_GIST_ID = "696115fe2df47fb2350fcff2663678c9"
372
 
373
- desc = """<center>💻 <a href="https://gist.github.com/chessllm">GitHub</a> | 🏆 <a href="https://gist.github.com/chessllm/696115fe2df47fb2350fcff2663678c9">Leaderboard</a> | ♟️ <a href="https://colab.research.google.com/drive/16ZELw3wWQ43aUFBH6f7GUhCm8HwPyhlY">Arena</a> | ♟️ <a href="https://colab.research.google.com/drive/1bHftO1h9lX9jFSnHJCKjycEeCY5khO5L">Trainer</a></center>"""
 
 
 
 
 
 
374
 
375
  with gr.Blocks() as demo:
376
- gr.Markdown("# ⚔️ Chess LLM Arena")
377
- gr.Markdown(desc)
378
  with gr.Row():
379
  model_id_white = gr.Textbox(label="White Model ID", value="mlabonne/chesspythia-70m")
380
- model_id_black = gr.Textbox(label="Black Model ID", value="EleutherAI/pythia-70m")
381
- btn = gr.Button("Run")
382
  with gr.Row():
383
  out = gr.Image(width=256)
384
  btn.click(fn=update, inputs=[model_id_white, model_id_black], outputs=out)
 
370
  RESULT_GIST_ID = "c491299e7b8a45a61ce5403a70cf8656"
371
  LEAD_GIST_ID = "696115fe2df47fb2350fcff2663678c9"
372
 
373
+ title = """
374
+ <div align="center">
375
+ <h1>⚔️ Chess LLM Arena</h1>
376
+ <h3>💻 <a href="https://gist.github.com/chessllm">GitHub</a> • 🏆 <a href="https://gist.github.com/chessllm/696115fe2df47fb2350fcff2663678c9">Leaderboard</a> • ♟️ <a href="https://colab.research.google.com/drive/16ZELw3wWQ43aUFBH6f7GUhCm8HwPyhlY">Arena</a> • ♟️ <a href="https://colab.research.google.com/drive/1bHftO1h9lX9jFSnHJCKjycEeCY5khO5L">Trainer</a></h3>
377
+ <p><em>Pick two chess LLMs and make them compete in a chess match. When the game is over, it will automatically update the crowd-sourced leaderboard.</em></p>
378
+ </div>
379
+ """
380
 
381
  with gr.Blocks() as demo:
382
+ gr.Markdown(title)
 
383
  with gr.Row():
384
  model_id_white = gr.Textbox(label="White Model ID", value="mlabonne/chesspythia-70m")
385
+ model_id_black = gr.Textbox(label="Black Model ID", value="BlueSunflower/Pythia-160M-chess")
386
+ btn = gr.Button("Fight!")
387
  with gr.Row():
388
  out = gr.Image(width=256)
389
  btn.click(fn=update, inputs=[model_id_white, model_id_black], outputs=out)