Update roop/processors/frame/face_enhancer.py

roop/processors/frame/face_enhancer.py

@@ -21,7 +21,7 @@ def get_face_enhancer() -> Any:
 
     with THREAD_LOCK:
         if FACE_ENHANCER is None:
-            model_path = resolve_relative_path('../models/GFPGANv1.pth')
+            model_path = resolve_relative_path('../models/GFPGANv1.4.onnx')
             # todo: set models path -> https://github.com/TencentARC/GFPGAN/issues/399
             FACE_ENHANCER = GFPGANer(model_path=model_path, upscale=1, device=get_device())
     return FACE_ENHANCER
@@ -43,7 +43,7 @@ def clear_face_enhancer() -> None:
 
 def pre_check() -> bool:
     download_directory_path = resolve_relative_path('../models')
-    conditional_download(download_directory_path, ['https://huggingface.co/TencentARC/GFPGANv1/resolve/main/GFPGANv1.pth'])
+    conditional_download(download_directory_path, ['https://huggingface.co/Neus/GFPGANv1.4/reslove/main/GFPGANv1.4.onnx'])
     return True