Update app.py

app.py

@@ -13,16 +13,22 @@ import spaces
 # ── Model loading ─────────────────────────────────────────────────────────────
 MODEL_ID = "lsadouk1111/VisualStep"
 
-print("Loading VisualStep model...")
-model, tokenizer = FastModel.from_pretrained(
-    model_name=MODEL_ID,
-    max_seq_length=512,
-    load_in_4bit=True,
-    dtype=None,
-)
-FastModel.for_inference(model)
-model.eval()
-print("Model ready.")
+model = None
+tokenizer = None
+
+def load_model():
+    global model, tokenizer
+    if model is None:
+        print("Loading VisualStep model...")
+        model, tokenizer = FastModel.from_pretrained(
+            model_name=MODEL_ID,
+            max_seq_length=512,
+            load_in_4bit=True,
+            dtype=None,
+        )
+        FastModel.for_inference(model)
+        model.eval()
+        print("Model ready.")
 
 # ── System prompt ─────────────────────────────────────────────────────────────
 SYSTEM_PROMPT = """You are an expert in ADHD classroom accommodations for primary school children aged 5-10.