tharms commited on
Commit
05c8de7
·
1 Parent(s): 8d74a31

added branding

Browse files
Files changed (1) hide show
  1. webui.py +2 -1
webui.py CHANGED
@@ -21,6 +21,7 @@ import torch
21
  from modules.sdxl_styles import legal_style_names
22
  from modules.private_logger import get_current_html_path
23
  from modules.ui_gradio_extensions import reload_javascript
 
24
  from modules.auth import auth_enabled, check_auth
25
  from modules.util import is_json
26
 
@@ -116,7 +117,7 @@ else:
116
  power_device = "CPU"
117
 
118
  with shared.gradio_root:
119
- gr.Markdown(f"""# <center>zStudios Generative Media AI running on {power_device}</center>""")
120
  with gr.Tab("DALL-E"):
121
  with gr.Row(variant="panel"):
122
  model = gr.Dropdown(choices=["dall-e-2", "dall-e-3"], label="Model", value="dall-e-3")
 
21
  from modules.sdxl_styles import legal_style_names
22
  from modules.private_logger import get_current_html_path
23
  from modules.ui_gradio_extensions import reload_javascript
24
+ from modules.ui_gradio_extensions import zStudiosHeader
25
  from modules.auth import auth_enabled, check_auth
26
  from modules.util import is_json
27
 
 
117
  power_device = "CPU"
118
 
119
  with shared.gradio_root:
120
+ gr.HTML(zStudiosHeader(power_device))
121
  with gr.Tab("DALL-E"):
122
  with gr.Row(variant="panel"):
123
  model = gr.Dropdown(choices=["dall-e-2", "dall-e-3"], label="Model", value="dall-e-3")