Clémentine commited on
Commit
a244751
1 Parent(s): de891db

add submission field

Browse files
Files changed (1) hide show
  1. src/submission/submit.py +19 -8
src/submission/submit.py CHANGED
@@ -75,6 +75,9 @@ def add_new_eval(
75
  architectures = getattr(model_config, "architectures", None)
76
  if architectures:
77
  architecture = ";".join(architectures)
 
 
 
78
 
79
 
80
  # Is the model info correctly filled?
@@ -95,23 +98,29 @@ def add_new_eval(
95
  if not modelcard_OK:
96
  return styled_error(error_msg)
97
 
 
 
 
 
98
  # Storing the model tags
99
  tags = []
100
-
101
- model_card = ModelCard.load(model)
102
- is_merge_from_metadata = False
103
  if model_card.data.tags:
104
- is_merge_from_metadata = "merge" in model_card.data.tags
105
- if "moe" in model_card.data.tags:
106
- tags.append("moe")
107
-
108
  merge_keywords = ["mergekit", "merged model", "merge model", "merging"]
109
  # If the model is a merge but not saying it in the metadata, we flag it
110
  is_merge_from_model_card = any(keyword in model_card.text.lower() for keyword in merge_keywords)
111
- if is_merge_from_model_card:
112
  tags.append("merge")
113
  if not is_merge_from_metadata:
114
  tags.append("flagged:undisclosed_merge")
 
 
 
 
 
 
 
115
 
116
 
117
  # Seems good, creating the eval
@@ -138,6 +147,8 @@ def add_new_eval(
138
  "license": license,
139
  "still_on_hub": True,
140
  "tags": tags,
 
 
141
  }
142
 
143
  # Check for duplicate submission
 
75
  architectures = getattr(model_config, "architectures", None)
76
  if architectures:
77
  architecture = ";".join(architectures)
78
+ downloads = getattr(model_config, 'downloads', 0)
79
+ created_at = getattr(model_config, 'created_at', '')
80
+
81
 
82
 
83
  # Is the model info correctly filled?
 
98
  if not modelcard_OK:
99
  return styled_error(error_msg)
100
 
101
+ is_merge_from_metadata = False
102
+ is_moe_from_metadata = False
103
+ model_card = ModelCard.load(model)
104
+
105
  # Storing the model tags
106
  tags = []
 
 
 
107
  if model_card.data.tags:
108
+ is_merge_from_metadata = "merge" in model_card.data.tags
109
+ is_moe_from_metadata = "moe" in model_card.data.tags
 
 
110
  merge_keywords = ["mergekit", "merged model", "merge model", "merging"]
111
  # If the model is a merge but not saying it in the metadata, we flag it
112
  is_merge_from_model_card = any(keyword in model_card.text.lower() for keyword in merge_keywords)
113
+ if is_merge_from_model_card or is_merge_from_metadata:
114
  tags.append("merge")
115
  if not is_merge_from_metadata:
116
  tags.append("flagged:undisclosed_merge")
117
+ moe_keywords = ["moe", "mixture of experts"]
118
+ is_moe_from_model_card = any(keyword in model_card.text.lower() for keyword in moe_keywords)
119
+ is_moe_from_name = "moe" in model.lower().replace("/", "-").replace("_", "-").split("-")
120
+ if is_moe_from_model_card or is_moe_from_name or is_moe_from_metadata:
121
+ tags.append("moe")
122
+ if not is_moe_from_metadata:
123
+ tags.append("flagged:undisclosed_moe")
124
 
125
 
126
  # Seems good, creating the eval
 
147
  "license": license,
148
  "still_on_hub": True,
149
  "tags": tags,
150
+ "downloads": downloads,
151
+ "created_at": created_at
152
  }
153
 
154
  # Check for duplicate submission