fix(api): get admin token from correct server object

ssube · Apr 22, 2023 · 6fecb1e · 6fecb1e
1 parent 4ac25f0
commit 6fecb1e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/api/entry.py b/api/entry.py
@@ -23,7 +23,7 @@ def script(obj, optimize=True, _frames_up=0, _rcb=None):
 
         # create the server and load the config
         from onnx_web.main import main
-        server, app, pool = main()
+        context, app, pool = main()
 
         # launch the image workers
         print("starting image workers")
@@ -39,7 +39,7 @@ def script(obj, optimize=True, _frames_up=0, _rcb=None):
         # launch the user's web browser
         print("opening web browser")
         url = "http://127.0.0.1:5000"
-        webbrowser.open_new_tab(f"{url}?api={url}&token={server.admin_token}")
+        webbrowser.open_new_tab(f"{url}?api={url}&token={context.admin_token}")
 
         # wait for enter and exit
         input("press enter to quit")