Views
No views yet
1from transformers import AutoModelForSequenceClassification, AutoTokenizer
2
3# โหลดโมเดลและ tokenizer
4model_name = "YOUR_USERNAME/thai-intent-classification"
5tokenizer = AutoTokenizer.from_pretrained(model_name)
6model = AutoModelForSequenceClassification.from_pretrained(model_name)
7
8# ทำนาย intent
9def predict_intent(text):
10 inputs = tokenizer(text, return_tensors="pt", truncation=True, padding=True)
11 outputs = model(**inputs)
12 predicted_class = outputs.logits.argmax().item()
13 intent_mapping = ["Scholarship", "academic_calendar", "course", "general_question", "student_activities"]
14 return intent_mapping[predicted_class]
15
16# ตัวอย่างการใช้งาน
17text = "ช่วงไหนเปิดเทอม"
18intent = predict_intent(text)
19print(f"Intent: {intent}")