load model function fix

akrsgm06 · Jun 21, 2023 · 29acad2 · 29acad2
1 parent 4ce41b6
commit 29acad2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/faceswap.py b/scripts/faceswap.py
@@ -18,8 +18,8 @@
 
 
 def get_models():
-    models = glob.glob(models_path)
     models_path = os.path.join(scripts.basedir(), "models/roop/*")
+    models = glob.glob(models_path)
     models += glob.glob(models_path)
     models = [x for x in models if x.endswith(".onnx") or x.endswith(".pth")]
     return models