vnyash
34 lines
1.3 KiB
Python
34 lines
1.3 KiB
Python
from typing import List, Optional
|
|
import gradio
|
|
import onnxruntime
|
|
|
|
import deepfuze.globals
|
|
from deepfuze import wording
|
|
from deepfuze.face_analyser import clear_face_analyser
|
|
from deepfuze.processors.frame.core import clear_frame_processors_modules
|
|
from deepfuze.execution import encode_execution_providers, decode_execution_providers
|
|
|
|
EXECUTION_PROVIDERS_CHECKBOX_GROUP : Optional[gradio.CheckboxGroup] = None
|
|
|
|
|
|
def render() -> None:
|
|
global EXECUTION_PROVIDERS_CHECKBOX_GROUP
|
|
|
|
EXECUTION_PROVIDERS_CHECKBOX_GROUP = gradio.CheckboxGroup(
|
|
label = wording.get('uis.execution_providers_checkbox_group'),
|
|
choices = encode_execution_providers(onnxruntime.get_available_providers()),
|
|
value = encode_execution_providers(deepfuze.globals.execution_providers)
|
|
)
|
|
|
|
|
|
def listen() -> None:
|
|
EXECUTION_PROVIDERS_CHECKBOX_GROUP.change(update_execution_providers, inputs = EXECUTION_PROVIDERS_CHECKBOX_GROUP, outputs = EXECUTION_PROVIDERS_CHECKBOX_GROUP)
|
|
|
|
|
|
def update_execution_providers(execution_providers : List[str]) -> gradio.CheckboxGroup:
|
|
clear_face_analyser()
|
|
clear_frame_processors_modules()
|
|
execution_providers = execution_providers or encode_execution_providers(onnxruntime.get_available_providers())
|
|
deepfuze.globals.execution_providers = decode_execution_providers(execution_providers)
|
|
return gradio.CheckboxGroup(value = execution_providers)
|