clean code, fix extract

This commit is contained in:
Tran Xen
2023-07-28 20:47:14 +02:00
parent 00d9cc6f62
commit a511214aaa
23 changed files with 328 additions and 194 deletions
+3 -2
View File
@@ -1,5 +1,6 @@
import glob
import os
from typing import List
import modules.scripts as scripts
from modules import scripts
from scripts.faceswaplab_globals import EXTENSION_PATH
@@ -7,7 +8,7 @@ from modules.shared import opts
from scripts.faceswaplab_utils.faceswaplab_logging import logger
def get_models():
def get_models() -> List[str]:
"""
Retrieve a list of swap model files.
@@ -44,7 +45,7 @@ def get_current_model() -> str:
return model
def get_face_checkpoints():
def get_face_checkpoints() -> List[str]:
"""
Retrieve a list of face checkpoint paths.