Spaces:
Running
Running
File size: 2,469 Bytes
6bcacf9 415da73 6bcacf9 415da73 6bcacf9 8c39361 6bcacf9 415da73 6bcacf9 8c39361 415da73 6bcacf9 415da73 6bcacf9 415da73 6bcacf9 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 |
import gradio as gr
from constants import APP_VERSION
from frontend.webui.text_to_image_ui import get_text_to_image_ui
from frontend.webui.image_to_image_ui import get_image_to_image_ui
from frontend.webui.generation_settings_ui import get_generation_settings_ui
from frontend.webui.models_ui import get_models_ui
from frontend.webui.image_variations_ui import get_image_variations_ui
from paths import FastStableDiffusionPaths
from state import get_settings
app_settings = get_settings()
def _get_footer_message() -> str:
version = f"<center><p> {APP_VERSION} "
footer_msg = version + (
' © 2023 <a href="https://github.com/rupeshs">'
" Rupesh Sreeraman</a></p></center>"
)
return footer_msg
def get_web_ui() -> gr.Blocks:
def change_mode(mode):
global app_settings
app_settings.settings.lcm_diffusion_setting.use_lcm_lora = False
app_settings.settings.lcm_diffusion_setting.use_openvino = False
if mode == "LCM-LoRA":
app_settings.settings.lcm_diffusion_setting.use_lcm_lora = True
elif mode == "LCM-OpenVINO":
app_settings.settings.lcm_diffusion_setting.use_openvino = True
with gr.Blocks(
css=FastStableDiffusionPaths.get_css_path(),
title="FastSD CPU",
) as fastsd_web_ui:
gr.HTML("<center><H1>FastSD CPU</H1></center>")
current_mode = "LCM"
if app_settings.settings.lcm_diffusion_setting.use_openvino:
current_mode = "LCM-OpenVINO"
elif app_settings.settings.lcm_diffusion_setting.use_lcm_lora:
current_mode = "LCM-LoRA"
mode = gr.Radio(
["LCM", "LCM-LoRA", "LCM-OpenVINO"],
label="Mode",
info="Current working mode",
value=current_mode,
)
mode.change(change_mode, inputs=mode)
with gr.Tabs():
with gr.TabItem("Text to Image"):
get_text_to_image_ui()
with gr.TabItem("Image to Image"):
get_image_to_image_ui()
with gr.TabItem("Image Variations"):
get_image_variations_ui()
with gr.TabItem("Generation Settings"):
get_generation_settings_ui()
with gr.TabItem("Models"):
get_models_ui()
gr.HTML(_get_footer_message())
return fastsd_web_ui
def start_webui(
share: bool = False,
):
webui = get_web_ui()
webui.launch(share=share)
|