From 0aae99e8082ab4bb6e6feb175663333308257b7f Mon Sep 17 00:00:00 2001 From: jh Date: Thu, 11 Mar 2021 23:41:12 -0800 Subject: [PATCH] fix: saved_models_path --- flaskr/app.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/flaskr/app.py b/flaskr/app.py index eb625a0..32a4f30 100644 --- a/flaskr/app.py +++ b/flaskr/app.py @@ -5,11 +5,11 @@ from flask_socketio import SocketIO, emit import logging -def create_flask_app(s2c, c2s, s2flask, args): +def create_flask_app(s2c, c2s, s2flask, kwargs): app = Flask(__name__, template_folder="templates", static_folder="static") log = logging.getLogger('werkzeug') log.disabled = True - model_path = Path(args.get('model_path', '')) + model_path = Path(kwargs.get('saved_models_path', '')) filename = 'preview.jpg' preview_file = str(model_path / filename)