Update app.py

app.py

@@ -3,15 +3,15 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
 
 MODEL_ID = "newtechdevng/math-tutor-smollm2-360M"
-BASE_ID  = "HuggingFaceTB/SmolLM2-360M-Instruct"  # tokenizer source
+BASE_ID  = "HuggingFaceTB/SmolLM2-360M-Instruct"
 
-print("Loading tokenizer from base model...")
-tokenizer = AutoTokenizer.from_pretrained(BASE_ID)  # ← fix: load from base
+print("Loading tokenizer...")
+tokenizer = AutoTokenizer.from_pretrained(BASE_ID)
 
-print("Loading fine-tuned model...")
+print("Loading model...")
 model = AutoModelForCausalLM.from_pretrained(
     MODEL_ID,
-    torch_dtype=torch.float32,
+    torch_dtype=torch.float16,
     device_map="cpu"
 )
 model.eval()
@@ -25,6 +25,7 @@ SYSTEM_PROMPT = (
 def solve(question, class_level, history):
     if not question.strip():
         return history, ""
+
     prompt = (
         "<|im_start|>system\n"
         + SYSTEM_PROMPT
@@ -45,10 +46,13 @@ def solve(question, class_level, history):
     reply = tokenizer.decode(outputs[0], skip_special_tokens=False)
     answer = reply.split("<|im_start|>assistant\n")[-1]
     answer = answer.replace("<|im_end|>", "").strip()
-    history.append((question, answer))
+
+    # ✅ Gradio 6 format — dict with role and content
+    history.append({"role": "user",      "content": question})
+    history.append({"role": "assistant", "content": answer})
     return history, ""
 
-with gr.Blocks(title="Math Tutor Class 6-10", theme=gr.themes.Soft()) as app:
+with gr.Blocks(title="Math Tutor Class 6-10") as app:
     gr.Markdown("# 🎓 Math Tutor — Class 6 to 10")
     gr.Markdown("Ask any math question and get step-by-step solutions!")
     with gr.Row():
@@ -56,7 +60,8 @@ with gr.Blocks(title="Math Tutor Class 6-10", theme=gr.themes.Soft()) as app:
             choices=["6", "7", "8", "9", "10"],
             value="8", label="Select Class", scale=1
         )
-    chatbot = gr.Chatbot(label="Math Solutions", height=450)
+    # ✅ Gradio 6 chatbot — type="messages"
+    chatbot = gr.Chatbot(label="Math Solutions", height=450, type="messages")
     with gr.Row():
         question = gr.Textbox(
             label="Your Question",