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
|