Spaces:
Runtime error
Runtime error
Update model_list.py
Browse files- model_list.py +3 -3
model_list.py
CHANGED
@@ -17,7 +17,6 @@ class ModelList:
|
|
17 |
<tr>
|
18 |
<td width="20%">Model Name</td>
|
19 |
<td width="10%">Type</td>
|
20 |
-
<td width="10%">Base Model</td>
|
21 |
<td width="10%">Paper</td>
|
22 |
<td width="10%">Code on Github</td>
|
23 |
<td width="10%">Weights on 🤗</td>
|
@@ -45,7 +44,6 @@ class ModelList:
|
|
45 |
<tr>
|
46 |
<td>{row.name}</td>
|
47 |
<td>{data_type}</td>
|
48 |
-
<td>{base_model}</td>
|
49 |
<td>{paper}</td>
|
50 |
<td>{github}</td>
|
51 |
<td>{hf_model}</td>
|
@@ -58,7 +56,8 @@ class ModelList:
|
|
58 |
case_sensitive: bool,
|
59 |
filter_names: list[str],
|
60 |
data_types: list[str],
|
61 |
-
model_types: list[str]
|
|
|
62 |
df = self.table
|
63 |
if search_query:
|
64 |
if case_sensitive:
|
@@ -69,6 +68,7 @@ class ModelList:
|
|
69 |
has_github = 'Code' in filter_names
|
70 |
has_model = 'Model Weights' in filter_names
|
71 |
df = self.filter_table(df, has_paper, has_github, has_model, data_types, model_types)
|
|
|
72 |
return len(df), self.to_html(df, self.table_header)
|
73 |
|
74 |
@staticmethod
|
|
|
17 |
<tr>
|
18 |
<td width="20%">Model Name</td>
|
19 |
<td width="10%">Type</td>
|
|
|
20 |
<td width="10%">Paper</td>
|
21 |
<td width="10%">Code on Github</td>
|
22 |
<td width="10%">Weights on 🤗</td>
|
|
|
44 |
<tr>
|
45 |
<td>{row.name}</td>
|
46 |
<td>{data_type}</td>
|
|
|
47 |
<td>{paper}</td>
|
48 |
<td>{github}</td>
|
49 |
<td>{hf_model}</td>
|
|
|
56 |
case_sensitive: bool,
|
57 |
filter_names: list[str],
|
58 |
data_types: list[str],
|
59 |
+
#model_types: list[str]
|
60 |
+
) -> tuple[int, str]:
|
61 |
df = self.table
|
62 |
if search_query:
|
63 |
if case_sensitive:
|
|
|
68 |
has_github = 'Code' in filter_names
|
69 |
has_model = 'Model Weights' in filter_names
|
70 |
df = self.filter_table(df, has_paper, has_github, has_model, data_types, model_types)
|
71 |
+
#df = self.filter_table(df, has_paper, has_github, has_model, data_types, model_types)
|
72 |
return len(df), self.to_html(df, self.table_header)
|
73 |
|
74 |
@staticmethod
|