Update app.py

app.py

@@ -22,6 +22,7 @@ import gradio as gr
 def generate_regex(board):
     """
     Generate regular expression for legal moves.
+    Based on https://gist.github.com/903124/cfbefa24da95e2316e0d5e8ef8ed360d by @903124S.
     """
     legal_moves = list(board.legal_moves)
     move_strings = [board.san(move) for move in legal_moves]
@@ -171,7 +172,9 @@ def save_pgn(final_pgn, file_name, auth_token):
 
 
 def download_file(base_url, file_name):
-    # Unique query parameter to bypass cache (using a timestamp)
+    """
+    Helper function to download a file.
+    """
     timestamp = datetime.now().strftime("%Y%m%d%H%M%S")
     url = f"{base_url}?ts={timestamp}"
 
@@ -191,6 +194,9 @@ def download_file(base_url, file_name):
 
 
 def get_leaderboard():
+    """
+    Wrapper to download the leaderboard and update the Gradio Dataframe.
+    """
     url = f"https://gist.githubusercontent.com/chessllm/{LEAD_GIST_ID}/raw"
     download_file(url, "chessllm_leaderboard.csv")
     return pd.read_csv("chessllm_leaderboard.csv")
@@ -281,7 +287,7 @@ def update(model_id_white, model_id_black):
 
         # Generate move
         start_time = time.time()
-        guided = generate.regex(current_model, regex_pattern, max_tokens=10)(prompt)
+        guided = generate.regex(current_model, regex_pattern)(prompt)
         end_time = time.time()
         move_duration = end_time - start_time
 
@@ -437,7 +443,7 @@ title = """
 </div>
 """
 
-with gr.Blocks() as demo:
+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")