Update app.py

app.py

@@ -3,18 +3,16 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
 
 MODEL_ID = "newtechdevng/math-tutor-smollm2-360M"
+BASE_ID  = "HuggingFaceTB/SmolLM2-360M-Instruct"  # tokenizer source
 
-print("Loading model...")
-tokenizer = AutoTokenizer.from_pretrained(
-    MODEL_ID,
-    use_fast=False,
-    trust_remote_code=True
-)
+print("Loading tokenizer from base model...")
+tokenizer = AutoTokenizer.from_pretrained(BASE_ID)  # ← fix: load from base
+
+print("Loading fine-tuned model...")
 model = AutoModelForCausalLM.from_pretrained(
     MODEL_ID,
     torch_dtype=torch.float32,
-    device_map="cpu",
-    trust_remote_code=True
+    device_map="cpu"
 )
 model.eval()
 print("Ready!")