reverted to 45f40f81c8878b7e31a535bea706873065c51367

webui.py

@@ -21,15 +21,10 @@ 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
 
-from fastapi import FastAPI
-from fastapi.responses import (HTMLResponse, JSONResponse)
-import uvicorn
-
-app = FastAPI()
-
 def ini_args():
     from args_manager import args
     return args
@@ -122,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")
@@ -759,7 +754,7 @@ dump_default_english_config()
 
 print(f'Starting Gradio... with {args_manager.args}')
 
-# shared.gradio_root.launch(debug=True)
+shared.gradio_root.launch(debug=True)
 """ shared.gradio_root.launch(
     inbrowser=args_manager.args.in_browser,
     server_name=args_manager.args.listen,
@@ -769,13 +764,4 @@ print(f'Starting Gradio... with {args_manager.args}')
     allowed_paths=[modules.config.path_outputs],
     blocked_paths=[constants.AUTH_FILENAME]
 )
- """
- 
-@app.get("/health", response_class=JSONResponse)
-def health():
-     return JSONResponse({'status': 'ok'})
-     
-app = gr.mount_gradio_app(app, shared.gradio_root, path="/")
-
-
-uvicorn.run(app)
+ """