added zStudio branding

modules/ui_gradio_extensions.py

@@ -52,6 +52,11 @@ def css_html():
     head = f'<link rel="stylesheet" property="stylesheet" href="{style_css_path}">'
     return head
 
+def zStudiosHeader(power_device):
+    logo_path = webpath('backstage.png')
+    zHead = f"""<table><tr><td><img src="{logo_path}" width="36" height="36"/></td><td><font size="5">zStudios Generative Media AI running on {power_device}</font></td></tr></table>"""
+    return zHead
+
 
 def reload_javascript():
     js = javascript_html()

webui.py

@@ -21,6 +21,7 @@ import torch
 from modules.sdxl_styles import legal_style_names
 from modules.private_logger import get_current_html_path
 from modules.ui_gradio_extensions import reload_javascript
+from modules.ui_gradio_extensions import zStudiosHeader
 from modules.auth import auth_enabled, check_auth
 from modules.util import is_json
 
@@ -116,7 +117,7 @@ else:
     power_device = "CPU"
 
 with shared.gradio_root:
-    gr.Markdown(f"""# <center>zStudios Generative Media AI running on {power_device}</center>""")
+    gr.HTML(zStudiosHeader(power_device))
     with gr.Tab("DALL-E"):
         with gr.Row(variant="panel"):
             model = gr.Dropdown(choices=["dall-e-2", "dall-e-3"], label="Model", value="dall-e-3")