update

app.py

@@ -209,14 +209,21 @@ def run_evaluation(text: str, progress=gr.Progress()):
     text = result  # Use cleaned text
 
     try:
-        # Step 1: Evaluate Qwen (using cached model)
-        progress(0, desc="Evaluating with Qwen3...")
-
-        # Get token count for prediction
+        # Get token counts for prediction first
         qwen_inputs = _qwen_tokenizer(text, return_tensors="pt", add_special_tokens=False)
         qwen_token_count = qwen_inputs["input_ids"].shape[-1]
         qwen_predicted_time = _stats_manager.predict_time("qwen", qwen_token_count)
 
+        rwkv_tokenized = _rwkv_tokenizer.encode(text)
+        rwkv_token_count = len(rwkv_tokenized.ids if hasattr(rwkv_tokenized, "ids") else rwkv_tokenized)
+        rwkv_predicted_time = _stats_manager.predict_time("rwkv", rwkv_token_count)
+
+        # Step 1: Evaluate Qwen (using cached model)
+        if qwen_predicted_time is not None:
+            progress(0, desc=f"Evaluating with Qwen3... (estimated: {qwen_predicted_time:.1f}s)")
+        else:
+            progress(0, desc="Evaluating with Qwen3...")
+
         result_qwen = evaluate_hf_single_sample(_qwen_model, _qwen_tokenizer, text, bos_mode="add_newline_token")
 
         # Save stats and print comparison
@@ -227,12 +234,10 @@ def run_evaluation(text: str, progress=gr.Progress()):
             print(f"Qwen3 completed in {result_qwen['inference_time']:.2f}s")
 
         # Step 2: Evaluate RWKV7 (using cached model)
-        progress(0, desc="Evaluating with RWKV7...")
-
-        # Get token count for prediction
-        rwkv_tokenized = _rwkv_tokenizer.encode(text)
-        rwkv_token_count = len(rwkv_tokenized.ids if hasattr(rwkv_tokenized, "ids") else rwkv_tokenized)
-        rwkv_predicted_time = _stats_manager.predict_time("rwkv", rwkv_token_count)
+        if rwkv_predicted_time is not None:
+            progress(0, desc=f"Evaluating with RWKV7... (estimated: {rwkv_predicted_time:.1f}s)")
+        else:
+            progress(0, desc="Evaluating with RWKV7...")
 
         result_rwkv = evaluate_rwkv7_single_sample(_rwkv_model, _rwkv_tokenizer, text)