diff --git a/app.py b/app.py index 95faf5d1..93eac5ac 100644 --- a/app.py +++ b/app.py @@ -310,10 +310,13 @@ async def on_connectionstatechange(): await pc.setLocalDescription(await pc.createOffer()) answer = await post(push_url,pc.localDescription.sdp) await pc.setRemoteDescription(RTCSessionDescription(sdp=answer,type='answer')) + + ########################################## # os.environ['MKL_SERVICE_FORCE_INTEL'] = '1' # os.environ['MULTIPROCESSING_METHOD'] = 'forkserver' -if __name__ == '__main__': +def main(): + global opt, model, avatar mp.set_start_method('spawn') parser = argparse.ArgumentParser() @@ -444,5 +447,7 @@ def run_server(runner): # print('start websocket server') # server = pywsgi.WSGIServer(('0.0.0.0', 8000), app, handler_class=WebSocketHandler) # server.serve_forever() - - + + +if __name__ == '__main__': + main()