diff --git a/backend/flaskApp.py b/backend/flaskApp.py index bc7a00f..9c1fb48 100644 --- a/backend/flaskApp.py +++ b/backend/flaskApp.py @@ -25,6 +25,7 @@ cfg.data.root_dir = str((CONFIG_PATH.parent / cfg.data.root_dir).resolve()) BASE_DIR = cfg.data.root_dir UPLOAD_DIR = cfg.data.upload_dir OUTPUT_DIR = cfg.data.run.output_dir +MODELS_DIR = str((CONFIG_PATH.parent / cfg.model.save_dir).resolve()) os.makedirs(UPLOAD_DIR, exist_ok=True) executor = ThreadPoolExecutor(max_workers=4) @@ -159,4 +160,9 @@ def preview(): "image": encoded }) - return jsonify({"ok": True, "count": len(result), "images": result}) \ No newline at end of file + return jsonify({"ok": True, "count": len(result), "images": result}) + +@app.get("/models") +def list_models(): + models_list = os.listdir(MODELS_DIR) + return jsonify({"ok": True, "models": models_list}) \ No newline at end of file diff --git a/frontend/index.html b/frontend/index.html index ed90868..2bea590 100644 --- a/frontend/index.html +++ b/frontend/index.html @@ -22,27 +22,21 @@