BearSean commited on
Commit
d054db2
1 Parent(s): 3cb49cd

Update model size regex (#19)

Browse files

- Update model size regex (203c648552275bb012789f32b56d0db1684b9bee)

src/display_models/get_model_metadata.py CHANGED
@@ -72,7 +72,7 @@ def get_model_likes(model_info):
72
  return model_info.likes
73
 
74
 
75
- size_pattern = re.compile(r"(\d\.)?\d+(b|m)")
76
 
77
 
78
  def get_model_size(model_name, model_info):
 
72
  return model_info.likes
73
 
74
 
75
+ size_pattern = re.compile(r"(\d+\.)?\d+(b|m)")
76
 
77
 
78
  def get_model_size(model_name, model_info):