Update app.py

app.py

@@ -61,14 +61,16 @@ except Exception:
 
 LEADERBOARD_DF = get_leaderboard_df(EVAL_RESULTS_PATH, EVAL_REQUESTS_PATH, COLS, BENCHMARK_COLS)
 
-# 모든 문자열 셀에 대해 benchmark 이름 변경 적용
-def replace_benchmark(s):
-    if isinstance(s, str):
-        s = s.replace("ANLI", "Korean Bar Exam (Lawyer)")
-        s = s.replace("LogiQA", "Senior Civil Service Examination(국가직 5급)")
-    return s
+# 변환 매핑 정의
+benchmark_mapping = {
+    "ANLI": "Korean Bar Exam (Lawyer)",
+    "LogiQA": "Senior Civil Service Examination(국가직 5급)"
+}
 
-LEADERBOARD_DF = LEADERBOARD_DF.applymap(replace_benchmark)
+# Leaderboard에 표시될 모델 타입 컬럼 변환 (컬럼 이름은 AutoEvalColumn.model_type.name로 가정)
+model_type_column = AutoEvalColumn.model_type.name
+if model_type_column in LEADERBOARD_DF.columns:
+    LEADERBOARD_DF[model_type_column] = LEADERBOARD_DF[model_type_column].apply(lambda s: benchmark_mapping.get(s, s))
 
 (
     finished_eval_queue_df,
@@ -107,13 +109,16 @@ def init_leaderboard(dataframe):
         interactive=False,
     )
 
-def transform_model_type(s):
-    # 만약 s 안에 "ANLI" 또는 "LogiQA"가 포함되어 있다면 변환 적용
-    if "ANLI" in s:
-        return s.replace("ANLI", "Korean Bar Exam (Lawyer)")
-    elif "LogiQA" in s:
-        return s.replace("LogiQA", "Senior Civil Service Examination(국가직 5급)")
-    return s
+def get_model_type_display(enum_obj):
+    """
+    ModelType enum 객체를 받아서 변환 매핑에 있으면 해당 값을 반환,
+    없으면 기본 to_str(" : ") 결과를 반환.
+    """
+    try:
+        key = enum_obj.name
+    except AttributeError:
+        key = enum_obj.to_str(" : ")
+    return benchmark_mapping.get(key, enum_obj.to_str(" : "))
 
 demo = gr.Blocks(css=custom_css)
 with demo:
@@ -175,7 +180,7 @@ with demo:
                     model_name_textbox = gr.Textbox(label="Model name")
                     revision_name_textbox = gr.Textbox(label="Revision commit", placeholder="main")
                     model_type = gr.Dropdown(
-                        choices=[transform_model_type(t.to_str(" : ")) for t in ModelType if t != ModelType.Unknown],
+                        choices=[get_model_type_display(t) for t in ModelType if t != ModelType.Unknown],
                         label="Model type",
                         multiselect=False,
                         value=None,