diff --git a/facefusion/app_context.py b/facefusion/app_context.py index d54f961e..8f7d404c 100644 --- a/facefusion/app_context.py +++ b/facefusion/app_context.py @@ -5,12 +5,14 @@ from facefusion.types import AppContext def detect_app_context() -> AppContext: + jobs_path = os.path.join('facefusion', 'jobs') + uis_path = os.path.join('facefusion', 'uis') frame = sys._getframe(1) while frame: - if os.path.join('facefusion', 'jobs') in frame.f_code.co_filename: + if jobs_path in frame.f_code.co_filename: return 'cli' - if os.path.join('facefusion', 'uis') in frame.f_code.co_filename: + if uis_path in frame.f_code.co_filename: return 'ui' frame = frame.f_back return 'cli'