Views
No views yet
| Metric | GPT-4o-mini | SFT Model | DPO Model |
|---|---|---|---|
| Classification accuracy | 74.8% | 79.9% | 78.0% |
| Order ID extraction | 98.7% | 99.4% | 99.4% |
| Language match rate | 96.2% | 99.4% | 100.0% |
1from transformers import AutoModelForCausalLM, AutoTokenizer
2import json
3
4model = AutoModelForCausalLM.from_pretrained("GuruVarshini/supportlm-qwen2.5-3b")
5tokenizer = AutoTokenizer.from_pretrained("GuruVarshini/supportlm-qwen2.5-3b")
6
7message = "bhai mere order ka kya hua yaar 3 din ho gaye"
8
9messages = [
10 {"role": "system", "content": "You are a customer support agent for an Indian e-commerce platform. Given a customer message, respond with a JSON object containing issue_type, order_id, and response."},
11 {"role": "user", "content": message}
12]
13
14input_ids = tokenizer.apply_chat_template(
15 messages,
16 return_tensors="pt",
17 add_generation_prompt=True
18)
19output = model.generate(input_ids, max_new_tokens=200)
20print(tokenizer.decode(output[0][input_ids.shape[1]:], skip_special_tokens=True))