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
|