fix: saved_models_path

This commit is contained in:
jh 2021-03-11 23:41:12 -08:00
commit 0aae99e808

View file

@ -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)