Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2import torch
3
4# Load model
5model = AutoModelForCausalLM.from_pretrained(
6 "ragib01/Qwen3-4B-customer-support",
7 torch_dtype=torch.float16,
8 device_map="auto",
9 trust_remote_code=True
10)
11
12tokenizer = AutoTokenizer.from_pretrained(
13 "ragib01/Qwen3-4B-customer-support",
14 trust_remote_code=True
15)
16
17# Test with a customer support query
18messages = [
19 {"role": "system", "content": "You are a helpful customer support assistant."},
20 {"role": "user", "content": "How do I track my order #74758657?"}
21]
22
23# Format input
24text = tokenizer.apply_chat_template(
25 messages,
26 tokenize=False,
27 add_generation_prompt=True
28)
29
30# Generate response
31inputs = tokenizer(text, return_tensors="pt").to(model.device)
32
33with torch.no_grad():
34 outputs = model.generate(
35 **inputs,
36 max_new_tokens=256,
37 temperature=0.7,
38 top_p=0.9,
39 do_sample=True
40 )
41
42response = tokenizer.decode(outputs[0], skip_special_tokens=True)
43print(response)1# Example: The model will generate tool calls for order tracking
2
3user_query = "Can you check the status of order #98765432?"
4
5# Model output will include:
6
7<tool_call>
8{
9 "name": "track_order",
10 "arguments": {
11 "order_number": "#98765432"
12 }
13}
14</tool_call>
15