openfree commited on
Commit
5a79046
·
verified ·
1 Parent(s): 3d591e6

Update src/populate.py

Browse files
Files changed (1) hide show
  1. src/populate.py +16 -0
src/populate.py CHANGED
@@ -60,3 +60,19 @@ def get_evaluation_queue_df(save_path: str, cols: list) -> list[pd.DataFrame]:
60
  df_running = pd.DataFrame.from_records(running_list, columns=cols)
61
  df_finished = pd.DataFrame.from_records(finished_list, columns=cols)
62
  return df_finished[cols], df_running[cols], df_pending[cols]
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
60
  df_running = pd.DataFrame.from_records(running_list, columns=cols)
61
  df_finished = pd.DataFrame.from_records(finished_list, columns=cols)
62
  return df_finished[cols], df_running[cols], df_pending[cols]
63
+
64
+ def make_clickable_model(model_name):
65
+ print("make_clickable_model 호출:", model_name) # 디버깅용 출력
66
+ if model_name in MODEL_MAPPING:
67
+ mapping_value = MODEL_MAPPING[model_name]
68
+ if isinstance(mapping_value, tuple):
69
+ new_name, new_url = mapping_value
70
+ else:
71
+ new_name = mapping_value
72
+ new_url = f"https://huggingface.co/{new_name}"
73
+ result = model_hyperlink(new_url, new_name)
74
+ else:
75
+ link = f"https://huggingface.co/{model_name}"
76
+ result = model_hyperlink(link, model_name)
77
+ print("변환 결과:", result) # 디버깅용 출력
78
+ return result