katielink commited on
Commit
f5e6909
·
1 Parent(s): d63a255

Update model_list.py

Browse files
Files changed (1) hide show
  1. 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]) -> tuple[int, 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