diff --git a/facefusion/predictor.py b/facefusion/predictor.py index 6eb30a08..0e2745f2 100644 --- a/facefusion/predictor.py +++ b/facefusion/predictor.py @@ -17,7 +17,7 @@ PREDICTOR = None THREAD_LOCK : threading.Lock = threading.Lock() NAME = 'FACEFUSION.PREDICTOR' MODEL_URL = 'https://github.com/facefusion/facefusion-assets/releases/download/models/open_nsfw.onnx' -MODEL_PATH = resolve_relative_path('../.assets/models/_open_nsfw.onnx') +MODEL_PATH = resolve_relative_path('../.assets/models/open_nsfw.onnx') MAX_PROBABILITY = 0.80 STREAM_COUNTER = 0 diff --git a/facefusion/processors/frame/modules/face_enhancer.py b/facefusion/processors/frame/modules/face_enhancer.py index 756c2bf8..792b8bbb 100644 --- a/facefusion/processors/frame/modules/face_enhancer.py +++ b/facefusion/processors/frame/modules/face_enhancer.py @@ -6,6 +6,7 @@ import numpy import onnxruntime import facefusion.globals +import facefusion.processors.frame.core as frame_processors from facefusion import wording from facefusion.face_analyser import get_many_faces, clear_face_analyser from facefusion.face_helper import warp_face, paste_back @@ -191,4 +192,4 @@ def process_image(source_path : str, target_path : str, output_path : str) -> No def process_video(source_path : str, temp_frame_paths : List[str]) -> None: - facefusion.processors.frame.core.multi_process_frames(None, temp_frame_paths, process_frames) + frame_processors.multi_process_frames(None, temp_frame_paths, process_frames)