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 |
|