Commit 34acad16 authored by DepFA's avatar DepFA Committed by AUTOMATIC1111
Browse files

Add GZipMiddleware to root demo

parent cc0258ae
Loading
Loading
Loading
Loading
+5 −1
Original line number Diff line number Diff line
@@ -5,6 +5,8 @@ import importlib
import signal
import threading

from fastapi.middleware.gzip import GZipMiddleware

from modules.paths import script_path

from modules import devices, sd_samplers
@@ -93,7 +95,7 @@ def webui():

        demo = modules.ui.create_ui(wrap_gradio_gpu_call=wrap_gradio_gpu_call)
        
        demo.launch(
        app,local_url,share_url = demo.launch(
            share=cmd_opts.share,
            server_name="0.0.0.0" if cmd_opts.listen else None,
            server_port=cmd_opts.port,
@@ -103,6 +105,8 @@ def webui():
            prevent_thread_lock=True
        )
        
        app.add_middleware(GZipMiddleware,minimum_size=1000)

        while 1:
            time.sleep(0.5)
            if getattr(demo, 'do_restart', False):