From 79cc0613fcbd38c66a2d2e4a61d38dda577184d7 Mon Sep 17 00:00:00 2001 From: henryruhs Date: Fri, 18 Oct 2024 11:52:09 +0200 Subject: [PATCH] Remove overrides for Gradio --- facefusion/uis/core.py | 6 ------ facefusion/uis/overrides.py | 15 --------------- 2 files changed, 21 deletions(-) delete mode 100644 facefusion/uis/overrides.py diff --git a/facefusion/uis/core.py b/facefusion/uis/core.py index 1dcd6715..360b860e 100644 --- a/facefusion/uis/core.py +++ b/facefusion/uis/core.py @@ -10,16 +10,10 @@ from gradio.themes import Size from facefusion import logger, metadata, state_manager, wording from facefusion.exit_helper import hard_exit from facefusion.filesystem import resolve_relative_path -from facefusion.uis import overrides from facefusion.uis.typing import Component, ComponentName os.environ['GRADIO_ANALYTICS_ENABLED'] = '0' -warnings.filterwarnings('ignore', category = UserWarning, module = 'gradio') - -gradio.processing_utils.encode_array_to_base64 = overrides.encode_array_to_base64 -gradio.processing_utils.encode_pil_to_base64 = overrides.encode_pil_to_base64 - UI_COMPONENTS: Dict[ComponentName, Component] = {} UI_LAYOUT_MODULES : List[ModuleType] = [] UI_LAYOUT_METHODS =\ diff --git a/facefusion/uis/overrides.py b/facefusion/uis/overrides.py deleted file mode 100644 index 1a1ee11d..00000000 --- a/facefusion/uis/overrides.py +++ /dev/null @@ -1,15 +0,0 @@ -import base64 -from typing import Any - -import cv2 -import numpy -from numpy._typing import NDArray - - -def encode_array_to_base64(array : NDArray[Any]) -> str: - _, buffer = cv2.imencode('.jpg', array[:, :, ::-1]) - return 'data:image/jpeg;base64,' + base64.b64encode(buffer.tobytes()).decode('utf-8') - - -def encode_pil_to_base64(image : Any) -> str: - return encode_array_to_base64(numpy.asarray(image)[:, :, ::-1])