add upload
Browse files- src/submission/submit.py +2 -3
src/submission/submit.py
CHANGED
|
@@ -3,6 +3,7 @@ import os
|
|
| 3 |
from datetime import datetime, timezone
|
| 4 |
|
| 5 |
from huggingface_hub import snapshot_download
|
|
|
|
| 6 |
from src.display.formatting import styled_error, styled_message, styled_warning
|
| 7 |
from src.envs import API, EVAL_REQUESTS_PATH, DYNAMIC_INFO_PATH, DYNAMIC_INFO_FILE_PATH, DYNAMIC_INFO_REPO, TOKEN, QUEUE_REPO, RATE_LIMIT_PERIOD, RATE_LIMIT_QUOTA
|
| 8 |
from src.submission.check_validity import (
|
|
@@ -93,9 +94,7 @@ def add_new_eval(
|
|
| 93 |
modelcard_OK, error_msg = check_model_card(model)
|
| 94 |
if not modelcard_OK:
|
| 95 |
return styled_error(error_msg)
|
| 96 |
-
|
| 97 |
-
# TODO unknown
|
| 98 |
-
tags = []
|
| 99 |
|
| 100 |
likes = model_info.likes
|
| 101 |
else:
|
|
|
|
| 3 |
from datetime import datetime, timezone
|
| 4 |
|
| 5 |
from huggingface_hub import snapshot_download
|
| 6 |
+
from src.submission.check_validity import get_model_tags
|
| 7 |
from src.display.formatting import styled_error, styled_message, styled_warning
|
| 8 |
from src.envs import API, EVAL_REQUESTS_PATH, DYNAMIC_INFO_PATH, DYNAMIC_INFO_FILE_PATH, DYNAMIC_INFO_REPO, TOKEN, QUEUE_REPO, RATE_LIMIT_PERIOD, RATE_LIMIT_QUOTA
|
| 9 |
from src.submission.check_validity import (
|
|
|
|
| 94 |
modelcard_OK, error_msg = check_model_card(model)
|
| 95 |
if not modelcard_OK:
|
| 96 |
return styled_error(error_msg)
|
| 97 |
+
tags = get_model_tags(model_card, model)
|
|
|
|
|
|
|
| 98 |
|
| 99 |
likes = model_info.likes
|
| 100 |
else:
|