ksort commited on
Commit
a6b754d
·
1 Parent(s): 533658a

Update ssh

Browse files
Files changed (2) hide show
  1. model/model_registry.py +2 -1
  2. serve/utils.py +1 -1
model/model_registry.py CHANGED
@@ -29,7 +29,8 @@ def get_model_description_md(model_list):
29
  ct = 0
30
  visited = set()
31
  for i, name in enumerate(model_list):
32
- minfo = get_model_info(name)
 
33
  if minfo.simple_name in visited:
34
  continue
35
  visited.add(minfo.simple_name)
 
29
  ct = 0
30
  visited = set()
31
  for i, name in enumerate(model_list):
32
+ model_source, model_name, model_type = name.split("_")
33
+ minfo = get_model_info(model_name)
34
  if minfo.simple_name in visited:
35
  continue
36
  visited.add(minfo.simple_name)
serve/utils.py CHANGED
@@ -176,7 +176,7 @@ def enable_vote_buttons():
176
  def disable_vote_buttons():
177
  return tuple(gr.update(visible=False, interactive=False) for i in range(6))
178
  def disable_vote():
179
- return (gr.update(interactive=False) for i in range(14))
180
  def enable_vote_mode_buttons(mode):
181
  print(mode)
182
  if mode == "Best":
 
176
  def disable_vote_buttons():
177
  return tuple(gr.update(visible=False, interactive=False) for i in range(6))
178
  def disable_vote():
179
+ return (gr.update(interactive=False) for i in range(18))
180
  def enable_vote_mode_buttons(mode):
181
  print(mode)
182
  if mode == "Best":