Update app.py

app.py

@@ -34,9 +34,9 @@ def json_to_row(path: str, metrics: dict) -> dict:
     precision = metrics.get("precision", "Unknown")
     model_type = metrics.get("model_type", "Unknown")
 
-    gsm8k_e2e = metrics.get("gsm8k_e2e_s", None)
-    gsm8k_bs = metrics.get("gsm8k_bs", None)
-    gsm8k_gpu = metrics.get("gpu_type", "")
+    e2e_s = metrics.get("e2e_s", None) 
+    batch_size = metrics.get("batch_size", None)
+    gpu_type = metrics.get("gpu_type", "")
 
     em = metrics.get("exact_match")
     correct = metrics.get("correct")
@@ -61,16 +61,16 @@ def json_to_row(path: str, metrics: dict) -> dict:
         "Method": method,
         "Model type": model_type,
         "Precision": precision,
-        "GSM8K<br>E2E(s)": f2(gsm8k_e2e),
-        "GSM8K<br>bs": gsm8k_bs,
-        "GSM8K<br>GPU": gsm8k_gpu,
-        "GSM8K<br>Accuracy(%)": pct(acc),
-        "GSM8K<br>Decoding T/s": f2(metrics.get("decoding_throughput")),
-        "GSM8K<br>Prefill T/s": f2(metrics.get("prefill_tp")),
-        "GSM8K<br>Prefill<br>S-MBU(%)": pct(metrics.get("prefill_smbu")),
-        "GSM8K<br>Prefill<br>S-MFU(%)": pct(metrics.get("prefill_smfu")),
-        "GSM8K<br>Decoding<br>S-MBU(%)": pct(metrics.get("decoding_smbu")),
-        "GSM8K<br>Decoding<br>S-MFU(%)": pct(metrics.get("decoding_smfu")),
+        "E2E(s)": f2(e2e_s),                  
+        "Batch size": batch_size,            
+        "GPU": gpu_type,                    
+        "Accuracy(%)": pct(acc),
+        "Decoding T/s": f2(metrics.get("decoding_throughput")),
+        "Prefill T/s": f2(metrics.get("prefill_tp")),
+        "Prefill<br>S-MBU(%)": pct(metrics.get("prefill_smbu")),
+        "Prefill<br>S-MFU(%)": pct(metrics.get("prefill_smfu")),
+        "Decoding<br>S-MBU(%)": pct(metrics.get("decoding_smbu")),
+        "Decoding<br>S-MFU(%)": pct(metrics.get("decoding_smfu")),
         "TTFT(s)": f2(metrics.get("ttft")),
         "TPOT(s)": f2(metrics.get("tpot")),
     }
@@ -198,11 +198,13 @@ def load_from_dir(
 
     # Model type filter
     if selected_model_types:
-        df = df[df["Model type"].isin(selected_model_types)]
+        lower_selected = [str(x).lower() for x in selected_model_types]
+        df = df[df["Model type"].astype(str).str.lower().isin(lower_selected)]
 
     # Precision filter
     if selected_precisions:
-        df = df[df["Precision"].isin(selected_precisions)]
+        lower_selected = [str(x).lower() for x in selected_precisions]
+        df = df[df["Precision"].astype(str).str.lower().isin(lower_selected)]
 
     if df.empty:
         empty_html = "<p>No records found.</p>"