Update src/populate.py
Browse files- 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
|