felix commited on
Commit
70693cf
·
1 Parent(s): 5f43914
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -47,7 +47,7 @@ hf_llm_diagrams = [img for img in imgs if 'hf_llm_diagram' in os.path.basename(i
47
  bigcode_diagrams = [img for img in imgs if 'bigcode' in os.path.basename(img)]
48
 
49
  # Getting the remaining images
50
- remaining_imgs = [img for img in imgs if 'hf_llm_diagram' not in os.path.basename(img)]
51
 
52
  def print_model_list(file_name, st, split_into_two=False):
53
  file_path = file_name[:-4] + '.json'
 
47
  bigcode_diagrams = [img for img in imgs if 'bigcode' in os.path.basename(img)]
48
 
49
  # Getting the remaining images
50
+ remaining_imgs = list(set(imgs) - set(hf_llm_diagrams) - set(bigcode_diagrams))
51
 
52
  def print_model_list(file_name, st, split_into_two=False):
53
  file_path = file_name[:-4] + '.json'