Update app.py

app.py

@@ -264,8 +264,8 @@ def update(model_id_white, model_id_black):
     result = None
 
     # Render first image
-    # image = render_init(board)
-    # yield image
+    image = render_init(board)
+    yield image
 
     # Time budget
     time_budget_white = TIME_BUDGET
@@ -445,19 +445,23 @@ title = """
   <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. Build a dataset and train your own small language model to compete in the arena.</em></p>
 </div>
 """
+footer = """
+<p><em>Made by Maxime Labonne, Kostis Gourgoulias, and Ruchi Bahl.</em></p>
+"""
 
 with gr.Blocks(theme=gr.themes.Monochrome()) as demo:
     gr.Markdown(title)
     with gr.Row():
-        model_id_white = gr.Textbox(label="White Model ID", value="mlabonne/chesspythia-70m")
-        model_id_black = gr.Textbox(label="Black Model ID", value="EleutherAI/pythia-70m-deduped")
+        model_id_white = gr.Textbox(label="♘ White Model ID", value="mlabonne/chesspythia-70m")
+        model_id_black = gr.Textbox(label="♞ Black Model ID", value="EleutherAI/pythia-70m-deduped")
     btn = gr.Button("Fight!")
     with gr.Row():
         gr.HTML("""<div id='chessboard' style='width: 50%; display: block;'></div>""")
         out = gr.Image(value=image, show_label=False, show_share_button=False, show_download_button=False, elem_id="chessboard")
         gr.HTML("""<div id='rightblock' style='width: 50%;  display: block;'></div>""")
     btn.click(fn=update, inputs=[model_id_white, model_id_black], outputs=out)
-    gr.Markdown('<div align="center"><p style="font-size: 36px;">🏆 Leaderboard</p></div>')
+    gr.Markdown('<div align="center"><p style="font-size: 30px;">🏆 Leaderboard</p></div>')
     leaderboard = gr.Dataframe(value=get_leaderboard, every=60)
+    gr.Markdown(footer)
 
 demo.queue(api_open=False).launch(show_api=False)