update

app.py

@@ -267,24 +267,7 @@ def run_evaluation(text: str, progress=gr.Progress()):
         # Wrap HTML for iframe display
         wrapped_html = wrap_html_in_iframe(html)
 
-        # Generate timing information for UI display
-        timing_lines = ["## ⏱️ Inference Timing\n"]
-
-        # Qwen timing
-        if qwen_predicted_time is not None:
-            timing_lines.append(f"**Qwen3-1.7B-Base**: {result_qwen['inference_time']:.2f}s (predicted: {qwen_predicted_time:.2f}s)")
-        else:
-            timing_lines.append(f"**Qwen3-1.7B-Base**: {result_qwen['inference_time']:.2f}s")
-
-        # RWKV timing
-        if rwkv_predicted_time is not None:
-            timing_lines.append(f"**RWKV7-G1C-1.5B**: {result_rwkv['inference_time']:.2f}s (predicted: {rwkv_predicted_time:.2f}s)")
-        else:
-            timing_lines.append(f"**RWKV7-G1C-1.5B**: {result_rwkv['inference_time']:.2f}s")
-
-        timing_text = "\n\n".join(timing_lines)
-
-        return wrapped_html, timing_text
+        return wrapped_html
 
     except torch.cuda.OutOfMemoryError:
         if torch.cuda.is_available():
@@ -300,7 +283,7 @@ def run_evaluation(text: str, progress=gr.Progress()):
 
 def clear_inputs():
     """Clear all inputs and outputs."""
-    return "", None, ""
+    return "", None
 
 
 def get_default_example():
@@ -309,9 +292,9 @@ def get_default_example():
 
     if _precomputed_html and _precomputed_text:
         wrapped_html = wrap_html_in_iframe(_precomputed_html)
-        return _precomputed_text, wrapped_html, ""
+        return _precomputed_text, wrapped_html
     else:
-        return "", None, ""
+        return "", None
 
 
 # Build Gradio UI
@@ -338,22 +321,17 @@ with gr.Blocks(title="Compression-Lens: RWKV-7 vs Qwen3", theme=gr.themes.Soft()
 
     gr.Markdown("---")
 
-    # Timing information display
-    with gr.Row():
-        with gr.Column():
-            timing_info = gr.Markdown(label="Inference Timing")
-
     with gr.Row():
         with gr.Column():
             output_html = gr.HTML(label="Visualization")
 
     # Event handlers
-    clear_btn.click(fn=clear_inputs, outputs=[text_input, output_html, timing_info])
+    clear_btn.click(fn=clear_inputs, outputs=[text_input, output_html])
 
-    run_btn.click(fn=run_evaluation, inputs=[text_input], outputs=[output_html, timing_info])
+    run_btn.click(fn=run_evaluation, inputs=[text_input], outputs=[output_html])
 
     # Load default example on page load
-    demo.load(fn=get_default_example, outputs=[text_input, output_html, timing_info])
+    demo.load(fn=get_default_example, outputs=[text_input, output_html])
 
 
 if __name__ == "__main__":