From b2c272c9946b6bae6d0c7edb9315d03af0c401a2 Mon Sep 17 00:00:00 2001
From: ebolam <ebolam@gmail.com>
Date: Thu, 21 Dec 2023 12:37:22 -0500
Subject: [PATCH] Add disable_model_load flag for classic UI. Only applies
 restriction on page refresh

---
 aiserver.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/aiserver.py b/aiserver.py
index 5d139431..5529d725 100644
--- a/aiserver.py
+++ b/aiserver.py
@@ -1907,7 +1907,7 @@ def index():
     if args.no_ui:
         return redirect('/api/latest')
     else:
-        return render_template('index.html', hide_ai_menu=args.noaimenu)
+        return render_template('index.html', hide_ai_menu=args.noaimenu or koboldai_vars.disable_model_load)
 @app.route('/api', strict_slashes=False)
 @require_allowed_ip
 def api():