| from typing import Optional |
|
|
| import gradio |
|
|
| from facefusion import wording |
| from facefusion.uis.components.benchmark import BENCHMARKS |
| from facefusion.uis.core import register_ui_component |
|
|
| BENCHMARK_RUNS_CHECKBOX_GROUP : Optional[gradio.CheckboxGroup] = None |
| BENCHMARK_CYCLES_SLIDER : Optional[gradio.Button] = None |
|
|
|
|
| def render() -> None: |
| global BENCHMARK_RUNS_CHECKBOX_GROUP |
| global BENCHMARK_CYCLES_SLIDER |
|
|
| BENCHMARK_RUNS_CHECKBOX_GROUP = gradio.CheckboxGroup( |
| label = wording.get('uis.benchmark_runs_checkbox_group'), |
| choices = list(BENCHMARKS.keys()), |
| value = list(BENCHMARKS.keys()) |
| ) |
| BENCHMARK_CYCLES_SLIDER = gradio.Slider( |
| label = wording.get('uis.benchmark_cycles_slider'), |
| value = 5, |
| step = 1, |
| minimum = 1, |
| maximum = 10 |
| ) |
| register_ui_component('benchmark_runs_checkbox_group', BENCHMARK_RUNS_CHECKBOX_GROUP) |
| register_ui_component('benchmark_cycles_slider', BENCHMARK_CYCLES_SLIDER) |
|
|
