Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Update model size regex
#19
by
BearSean
- opened
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
|
| 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):
|