Spaces:
Runtime error
Runtime error
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
|