admin commited on
Commit
2f89051
·
1 Parent(s): 3a6750c

pt_files.insert(0, model)

Files changed (1) hide show
  1. utils.py +2 -2
utils.py CHANGED
@@ -36,9 +36,9 @@ def get_modelist(model_dir=MODEL_DIR, assign_model=""):
36
  if file.endswith(".pt"):
37
  model = os.path.basename(file)[:-3]
38
  if assign_model and assign_model.lower() in model:
39
- pt_files.append(model)
40
- else:
41
  pt_files.insert(0, model)
 
 
42
 
43
  return pt_files
44
 
 
36
  if file.endswith(".pt"):
37
  model = os.path.basename(file)[:-3]
38
  if assign_model and assign_model.lower() in model:
 
 
39
  pt_files.insert(0, model)
40
+ else:
41
+ pt_files.append(model)
42
 
43
  return pt_files
44