Views
No views yet
1from transformers import pipeline
2
3# Charger le modèle
4classifier = pipeline("text-classification", model="your-username/distilbert-query-classifier")
5
6# Classifier une requête
7result = classifier("What are the prerequisites for the machine learning course?")
8print(result)
9# [{'label': 'question_rag', 'score': 0.92}]1queries = [
2 "What topics are covered in the Python course?", # → question_rag
3 "Send a message to John about the meeting", # → send_message
4]
5
6results = classifier(queries)