Skip to content

S
Stable Diffusion Webui
Commit afaa03c5 authored by DepFA's avatar DepFA Committed by GitHub
Browse files
Options

add redefinition guard to gradio_routes_templates_response

parent a9044475
Hide whitespace changes
Inline Side-by-side
Showing with 8 additions and 7 deletions
modules/ui.py
View file @ afaa03c5
...@@ -1219,12 +1219,13 @@ for filename in sorted(os.listdir(jsdir)): ...@@ -1219,12 +1219,13 @@ for filename in sorted(os.listdir(jsdir)):
javascript += f"\n<script>{jsfile.read()}</script>" javascript += f"\n<script>{jsfile.read()}</script>"
def template_response(*args, **kwargs): if 'gradio_routes_templates_response' not in globals():
res = gradio_routes_templates_response(*args, **kwargs) def template_response(*args, **kwargs):
res.body = res.body.replace(b'</head>', f'{javascript}</head>'.encode("utf8")) res = gradio_routes_templates_response(*args, **kwargs)
res.init_headers() res.body = res.body.replace(b'</head>', f'{javascript}</head>'.encode("utf8"))
return res res.init_headers()
return res
gradio_routes_templates_response = gradio.routes.templates.TemplateResponse
gradio.routes.templates.TemplateResponse = template_response
gradio_routes_templates_response = gradio.routes.templates.TemplateResponse
gradio.routes.templates.TemplateResponse = template_response
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment