Create app.py

app.py

@@ -0,0 +1,76 @@
+import gradio as gr
+from transformers import AutoTokenizer, AutoModelForCausalLM
+import torch
+
+print("Loading model...")
+MODEL_ID = "newtechdevng/qwen-math-tutor"
+tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
+model = AutoModelForCausalLM.from_pretrained(
+    MODEL_ID,
+    torch_dtype = torch.float32,
+    device_map  = "cpu"
+)
+print("Model loaded!")
+
+def solve(question, history):
+    messages = [
+        {"role": "system", "content": "You are a helpful math tutor. Solve problems step by step, showing all working clearly."},
+    ]
+    for h in history:
+        messages.append({"role": "user",      "content": h[0]})
+        messages.append({"role": "assistant", "content": h[1]})
+    messages.append({"role": "user", "content": question})
+
+    text = tokenizer.apply_chat_template(
+        messages, tokenize=False, add_generation_prompt=True
+    )
+    inputs = tokenizer(text, return_tensors="pt")
+
+    with torch.no_grad():
+        out = model.generate(
+            **inputs,
+            max_new_tokens = 512,
+            temperature    = 0.1,
+            do_sample      = True,
+            pad_token_id   = tokenizer.eos_token_id,
+        )
+    return tokenizer.decode(
+        out[0][inputs["input_ids"].shape[1]:],
+        skip_special_tokens=True
+    )
+
+with gr.Blocks(theme=gr.themes.Soft(), title="Math Tutor AI") as demo:
+    gr.Markdown("""
+    # 🧮 Math Tutor AI
+    ### Powered by Qwen2.5-Math — Fine-tuned for NCERT & competitive math
+    """)
+
+    chatbot = gr.Chatbot(height=400, label="Math Tutor")
+
+    with gr.Row():
+        question = gr.Textbox(
+            placeholder = "Type your math question here...",
+            label       = "Your Question",
+            scale       = 4
+        )
+        submit = gr.Button("Solve →", variant="primary", scale=1)
+
+    gr.Examples(
+        examples=[
+            "Solve: 2x² - 7x + 3 = 0",
+            "Find the area of a triangle with base 12 cm and height 8 cm.",
+            "If sin θ = 3/5, find cos θ and tan θ.",
+            "A train travels 360 km in 4 hours. Find its speed in m/s.",
+        ],
+        inputs=question
+    )
+
+    def respond(q, history):
+        answer = solve(q, history)
+        history.append((q, answer))
+        return "", history
+
+    submit.click(respond, [question, chatbot], [question, chatbot])
+    question.submit(respond, [question, chatbot], [question, chatbot])
+
+demo.launch()