charliebaby2023 commited on
Commit
ada75fc
·
verified ·
1 Parent(s): e500098

Update frontend/webui/ui.py

Browse files
Files changed (1) hide show
  1. frontend/webui/ui.py +22 -20
frontend/webui/ui.py CHANGED
@@ -15,11 +15,8 @@ app_settings = get_settings()
15
  model_list = ["model-a", "model-b", "model-c"]
16
 
17
  def update_model_array(text_input):
18
- # Split lines, remove empty strings, strip whitespace
19
  updated = [line.strip() for line in text_input.split("\n") if line.strip()]
20
- return gr.Dropdown.update(choices=updated, value=updated[0] if updated else None), "\n".join(updated)
21
-
22
-
23
 
24
 
25
  def _get_footer_message() -> str:
@@ -46,22 +43,27 @@ def get_web_ui() -> gr.Blocks:
46
  title="FastSD CPU",
47
  ) as fastsd_web_ui:
48
  gr.HTML("<center><H1>FastSD CPU</H1></center>")
49
- current_mode = "LCM"
50
- if app_settings.settings.lcm_diffusion_setting.use_openvino:
51
- current_mode = "LCM-OpenVINO"
52
- elif app_settings.settings.lcm_diffusion_setting.use_lcm_lora:
53
- current_mode = "LCM-LoRA"
54
- model_dropdown = gr.Dropdown(choices=model_list, label="Select Model", value=model_list[0])
55
- model_text_area = gr.Textbox(value="\n".join(model_list), lines=10, label="Edit Model List")
56
- update_button = gr.Button("Update Dropdown")
57
- update_button.click( fn=update_model_array, inputs=model_text_area, outputs=[model_dropdown, model_text_area], )
58
- mode = gr.Radio(
59
- ["LCM", "LCM-LoRA", "LCM-OpenVINO"],
60
- label="Mode",
61
- info="Current working mode",
62
- value=current_mode,
63
- )
64
- mode.change(change_mode, inputs=mode)
 
 
 
 
 
65
 
66
  with gr.Tabs():
67
  with gr.TabItem("Text to Image"):
 
15
  model_list = ["model-a", "model-b", "model-c"]
16
 
17
  def update_model_array(text_input):
 
18
  updated = [line.strip() for line in text_input.split("\n") if line.strip()]
19
+ return gr.update(choices=updated, value=updated[0] if updated else None), "\n".join(updated)
 
 
20
 
21
 
22
  def _get_footer_message() -> str:
 
43
  title="FastSD CPU",
44
  ) as fastsd_web_ui:
45
  gr.HTML("<center><H1>FastSD CPU</H1></center>")
46
+ gr.Column(scale=1):
47
+ current_mode = "LCM"
48
+ if app_settings.settings.lcm_diffusion_setting.use_openvino:
49
+ current_mode = "LCM-OpenVINO"
50
+ elif app_settings.settings.lcm_diffusion_setting.use_lcm_lora:
51
+ current_mode = "LCM-LoRA"
52
+ gr.Column(scale=1):
53
+ gr.Row():
54
+ model_text_area2 = gr.Textbox(value=" ", lines=10, label="Enter a new model")
55
+ update_button = gr.Button("Update Dropdown")
56
+ update_button.click( fn=update_model_array, inputs=model_text_area2, outputs=[model_dropdown, model_text_area], )
57
+ gr.Row():
58
+ model_dropdown = gr.Dropdown(choices=model_list, label="Select Model", value=model_list[0])
59
+ model_text_area = gr.Textbox(value="\n".join(model_list), lines=10, label="Edit Model List")
60
+ mode = gr.Radio(
61
+ ["LCM", "LCM-LoRA", "LCM-OpenVINO"],
62
+ label="Mode",
63
+ info="Current working mode",
64
+ value=current_mode,
65
+ )
66
+ mode.change(change_mode, inputs=mode)
67
 
68
  with gr.Tabs():
69
  with gr.TabItem("Text to Image"):