Update app.py

app.py

@@ -444,63 +444,6 @@ def normalize_parameter_size(param_size: str) -> str:
     except:
         return param_size  # Return original if conversion fails
 
-def load_latest_model_stats():
-    """Load model stats from the model_stats.json file."""
-    try:
-        # Read directly from model_stats.json in root directory
-        with open('model_stats.json', 'r') as f:
-            stats = json.load(f)
-            
-        # Convert stats to table format
-        table_data = []
-        headers = ["Model", "VRAM (GB)", "Size", "Parameters", "Quantization", "Tokens/sec", "Gen Tokens/sec", "Total Tokens", "Response Time (s)"]
-        
-        for model in stats:
-            if not model.get("success", False):  # Skip failed tests
-                continue
-                
-            perf = model.get("performance", {})
-            info = model.get("model_info", {})
-            
-            try:
-                # Format numeric values with 2 decimal places
-                model_size = float(info.get("size", 0))  # Get raw size
-                vram_gb = round(model_size/1024/1024/1024, 2)  # Convert to GB
-                tokens_per_sec = round(float(perf.get("tokens_per_second", 0)), 2)
-                gen_tokens_per_sec = round(float(perf.get("generation_tokens_per_second", 0)), 2)
-                total_tokens = perf.get("total_tokens", 0)
-                response_time = round(float(perf.get("response_time", 0)), 2)
-                
-                # Normalize parameter size to billions format
-                param_size = normalize_parameter_size(info.get("parameter_size", "Unknown"))
-                
-                row = [
-                    model.get("model_name", "Unknown"),      # String
-                    vram_gb,                                 # Number (2 decimals)
-                    model_size,                              # Number (bytes)
-                    param_size,                              # String (normalized to B)
-                    info.get("quantization_level", "Unknown"),  # String
-                    tokens_per_sec,                          # Number (2 decimals)
-                    gen_tokens_per_sec,                      # Number (2 decimals)
-                    total_tokens,                            # Number (integer)
-                    response_time                            # Number (2 decimals)
-                ]
-                table_data.append(row)
-            except Exception as row_error:
-                logger.warning(f"Skipping model {model.get('model_name', 'Unknown')}: {str(row_error)}")
-                continue
-            
-        if not table_data:
-            return None, "No valid model stats found"
-            
-        # Sort by tokens per second (numerically)
-        table_data.sort(key=lambda x: float(x[5]) if isinstance(x[5], (int, float)) else 0, reverse=True)
-        
-        return headers, table_data
-    except Exception as e:
-        logger.error(f"Error in load_latest_model_stats: {str(e)}")
-        return None, f"Error loading model stats: {str(e)}"
-
 # Initialize Gradio Blocks
 with gr.Blocks(css="""
     #dice-button {
@@ -575,7 +518,7 @@ with gr.Blocks(css="""
     with gr.Tab("Battle Arena"):
         with gr.Row():
             prompt_input = gr.Textbox(
-                label="Enter your prompt", 
+                label="Enter your prompt",
                 placeholder="Type your prompt here...",
                 scale=20
             )
@@ -601,8 +544,8 @@ with gr.Blocks(css="""
             right_vote_btn = gr.Button(f"Vote for {right_output.label}", interactive=False)
         
         result = gr.Textbox(
-            label="Status", 
-            interactive=False, 
+            label="Status",
+            interactive=False,
             value="Generate responses to start the battle! 🚀",
             visible=True  # Always visible
         )
@@ -642,38 +585,12 @@ with gr.Blocks(css="""
             outputs=[release_notes]
         )
     
-    # Model Stats Tab
-    with gr.Tab("Model Stats"):
-        gr.Markdown("""
-        ### Model Performance Statistics
-        
-        This tab shows detailed performance metrics for each model, tested using a creative writing prompt.
-        The tests were performed on an **AMD Radeon RX 7600 XT 16GB GPU**.
-        
-        For detailed information about the testing methodology, parameters, and hardware setup, please refer to the 
-        [README_model_stats.md](https://huggingface.co/spaces/k-mktr/gpu-poor-llm-arena/blob/main/README_model_stats.md).
-        
-        """)
-        
-        headers, table_data = load_latest_model_stats()
-        if headers:
-            model_stats_table = gr.Dataframe(
-                headers=headers,
-                value=table_data,
-                row_count=len(table_data),
-                col_count=len(headers),
-                interactive=True,
-                label="Model Performance Statistics"
-            )
-        else:
-            gr.Markdown(f"⚠️ {table_data}")  # Show error message if loading failed
-    
     # Define interactions
     submit_btn.click(
         battle_arena,
         inputs=prompt_input,
         outputs=[
-            left_output, right_output, left_model, right_model, 
+            left_output, right_output, left_model, right_model,
             left_output, right_output, left_vote_btn, right_vote_btn,
             tie_btn, previous_prompt, tie_count, model_names_row, result
         ]
@@ -682,14 +599,14 @@ with gr.Blocks(css="""
     left_vote_btn.click(
         lambda *args: record_vote(*args, "Left is better"),
         inputs=[prompt_input, left_output, right_output, left_model, right_model],
-        outputs=[result, leaderboard, elo_leaderboard, left_vote_btn, 
+        outputs=[result, leaderboard, elo_leaderboard, left_vote_btn,
                  right_vote_btn, tie_btn, model_names_row]
     )
     
     right_vote_btn.click(
         lambda *args: record_vote(*args, "Right is better"),
         inputs=[prompt_input, left_output, right_output, left_model, right_model],
-        outputs=[result, leaderboard, elo_leaderboard, left_vote_btn, 
+        outputs=[result, leaderboard, elo_leaderboard, left_vote_btn,
                  right_vote_btn, tie_btn, model_names_row]
     )
     
@@ -701,7 +618,7 @@ with gr.Blocks(css="""
     
     new_battle_btn.click(
         new_battle,
-        outputs=[prompt_input, left_output, right_output, left_model, 
+        outputs=[prompt_input, left_output, right_output, left_model,
                 right_model, left_vote_btn, right_vote_btn, tie_btn,
                 result, leaderboard, model_names_row, elo_leaderboard, tie_count]
     )