optimize(web): program cleanup on exit

This commit is contained in:
源文雨 2024-06-04 15:45:17 +09:00
parent 0f36233017
commit bd0b43f7e6

View File

@ -1638,16 +1638,24 @@ with gr.Blocks(title="RVC WebUI") as app:
except: except:
gr.Markdown(traceback.format_exc()) gr.Markdown(traceback.format_exc())
try: try:
if config.iscolab: import signal
app.queue(max_size=1022).launch(share=True, max_threads=511) def cleanup(signum, frame):
else: signame = signal.Signals(signum).name
app.queue(max_size=1022).launch( print(f'Got signal {signame} ({signum})')
max_threads=511, app.close()
server_name="0.0.0.0", sys.exit(0)
inbrowser=not config.noautoopen, signal.signal(signal.SIGINT, cleanup)
server_port=config.listen_port, signal.signal(signal.SIGTERM, cleanup)
quiet=True, if config.iscolab:
) app.queue(max_size=1022).launch(share=True, max_threads=511)
except Exception as e: else:
logger.error(str(e)) app.queue(max_size=1022).launch(
max_threads=511,
server_name="0.0.0.0",
inbrowser=not config.noautoopen,
server_port=config.listen_port,
quiet=True,
)
except Exception as e:
logger.error(str(e))