优雅退出

server.py

@@ -234,7 +234,16 @@ def get_tasks(tasks:list=[]):
             time.sleep(2)
     return tasks
 
+import sys                                                                  
+import signal
+
+def quit(signum, frame):
+    print('stop fusion')
+    sys.exit()
+
 def main_dev(opt):
+    signal.signal(signal.SIGINT, quit)
+    signal.signal(signal.SIGTERM, quit)
     model_name = '' # 默认模型
     model = None    # 默认模型
     config = None   # 默认配置