Spaces:
Running
Running
admin
commited on
Commit
·
2f89051
1
Parent(s):
3a6750c
fix
Browse filespt_files.insert(0, model)
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 |
|