diff --git a/.gitignore b/.gitignore index e94022a..c336858 100644 --- a/.gitignore +++ b/.gitignore @@ -7,3 +7,4 @@ build /api/db.sqlite3 /emblemscanner-release models +emblem5/ai/data/ diff --git a/emblem5/ai/server.py b/emblem5/ai/server.py index 104c230..676039e 100755 --- a/emblem5/ai/server.py +++ b/emblem5/ai/server.py @@ -83,7 +83,8 @@ def find_best_frame(files): def do_qr_verify(): start_time = time.time() fd = flask.request.form - model_path, model_md5 = download_model(fd.get('model', default_model)) + model_name = fd.get('model', default_model) + model_path, model_md5 = download_model(model_name) model, transforms = load_model(model_path) frame_image, clarities = find_best_frame(flask.request.files) @@ -110,6 +111,7 @@ def do_qr_verify(): "probabilities": ', '.join([f'{v:.2%}' for k, v in probabilities]), "clarities": clarities, "model_md5": model_md5, + "model_name": model_name, } }