Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model = AutoModelForCausalLM.from_pretrained("Shivam250124/socrates-tutor-qwen-1.5b")
4tokenizer = AutoTokenizer.from_pretrained("Shivam250124/socrates-tutor-qwen-1.5b")
5
6prompt = '''System: You are a Socratic tutor. Ask questions, never give answers.
7
8User: I don't understand why 0.1 + 0.2 != 0.3 in Python.
9
10What is your Socratic question?'''
11
12inputs = tokenizer(prompt, return_tensors="pt")
13outputs = model.generate(**inputs, max_new_tokens=100)
14print(tokenizer.decode(outputs[0], skip_special_tokens=True))| Task | Baseline | Trained | Improvement |
|---|---|---|---|
| Easy | -0.15 | +0.85 | 467% |
| Medium | -0.45 | +0.60 | 233% |
| Hard | -0.80 | +0.45 | 156% |