Update scripts/swap.py
Browse files- scripts/swap.py +1 -1
scripts/swap.py
CHANGED
|
@@ -22,6 +22,7 @@ def setup_models(execution_provider="cpu"):
|
|
| 22 |
|
| 23 |
if enhancer is None:
|
| 24 |
enhancer = GFPGANer(
|
|
|
|
| 25 |
upscale=1,
|
| 26 |
arch="clean",
|
| 27 |
channel_multiplier=2,
|
|
@@ -57,4 +58,3 @@ def run_swap(target_path, source_path, output_path, execution_provider="cpu", fr
|
|
| 57 |
|
| 58 |
cv2.imwrite(output_path, target_img)
|
| 59 |
return True
|
| 60 |
-
|
|
|
|
| 22 |
|
| 23 |
if enhancer is None:
|
| 24 |
enhancer = GFPGANer(
|
| 25 |
+
model_path="https://github.com/TencentARC/GFPGAN/releases/download/v1.3.8/GFPGANv1.4.pth",
|
| 26 |
upscale=1,
|
| 27 |
arch="clean",
|
| 28 |
channel_multiplier=2,
|
|
|
|
| 58 |
|
| 59 |
cv2.imwrite(output_path, target_img)
|
| 60 |
return True
|
|
|