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
@@ -434,6 +434,9 @@ url1 = (f"https://gist.githubusercontent.com/chessllm/{RESULT_GIST_ID}/raw")
 download_file(url1, "chessllm_results.csv")
 elo_ratings_df = get_leaderboard()
 
+# Render chessboard
+board = chess.Board()
+image = render_init(board)
 
 title = """
 <div align="center">
@@ -451,7 +454,7 @@ with gr.Blocks(theme=gr.themes.Monochrome()) as demo:
     btn = gr.Button("Fight!")
     with gr.Row():
         gr.HTML("""<div id='chessboard' style='width: 50%; display: block;'></div>""")
-        out = gr.Image(show_label=False, show_share_button=False, show_download_button=False, elem_id="chessboard")
+        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>')