Update src/submission/submit.py

#52
Files changed (1) hide show
  1. src/submission/submit.py +4 -0
src/submission/submit.py CHANGED
@@ -81,6 +81,10 @@ def add_new_eval(
81
 
82
  model_size = get_model_size(model_info=model_info, precision=precision)
83
 
 
 
 
 
84
  # Were the model card and license filled?
85
  try:
86
  license = model_info.cardData["license"]
 
81
 
82
  model_size = get_model_size(model_info=model_info, precision=precision)
83
 
84
+ # Were the model size less than 30B?
85
+ if float(model_size) >= 30:
86
+ return styled_error("Please submit a model that is less than 30B")
87
+
88
  # Were the model card and license filled?
89
  try:
90
  license = model_info.cardData["license"]