Spaces:
Runtime error
Runtime error
Update model_list.py
Browse files- model_list.py +5 -2
model_list.py
CHANGED
|
@@ -73,7 +73,10 @@ class ModelList:
|
|
| 73 |
|
| 74 |
@staticmethod
|
| 75 |
def filter_table(df: pd.DataFrame, has_paper: bool, has_github: bool,
|
| 76 |
-
has_model: bool,
|
|
|
|
|
|
|
|
|
|
| 77 |
if has_paper:
|
| 78 |
df = df[~df.paper.isna()]
|
| 79 |
if has_github:
|
|
@@ -81,7 +84,7 @@ class ModelList:
|
|
| 81 |
if has_model:
|
| 82 |
df = df[~df.hub.isna() | ~df.other.isna()]
|
| 83 |
df = df[df.data_type.isin(set(data_types))]
|
| 84 |
-
df = df[df.base_model.isin(set(model_types))]
|
| 85 |
return df
|
| 86 |
|
| 87 |
@staticmethod
|
|
|
|
| 73 |
|
| 74 |
@staticmethod
|
| 75 |
def filter_table(df: pd.DataFrame, has_paper: bool, has_github: bool,
|
| 76 |
+
has_model: bool,
|
| 77 |
+
data_types: list[str],
|
| 78 |
+
#model_types: list[str]
|
| 79 |
+
) -> pd.DataFrame:
|
| 80 |
if has_paper:
|
| 81 |
df = df[~df.paper.isna()]
|
| 82 |
if has_github:
|
|
|
|
| 84 |
if has_model:
|
| 85 |
df = df[~df.hub.isna() | ~df.other.isna()]
|
| 86 |
df = df[df.data_type.isin(set(data_types))]
|
| 87 |
+
#df = df[df.base_model.isin(set(model_types))]
|
| 88 |
return df
|
| 89 |
|
| 90 |
@staticmethod
|