Views
No views yet
1intent_classes = {
2 0: 'abroad',
3 1: 'address',
4 2: 'app_error',
5 3: 'atm_limit',
6 4: 'balance',
7 5: 'business_loan',
8 6: 'card_issues',
9 7: 'cash_deposit',
10 8: 'direct_debit',
11 9: 'freeze',
12 10: 'high_value_payment',
13 11: 'joint_account',
14 12: 'latest_transactions',
15 13: 'pay_bill'
16}1>>> from transformers import pipeline
2
3model = "/content/RoBERTa-mind14-classifier-intent"
4classifier = pipeline("text-classification", model=model)
5
6text = "hi what's the maximum amount of money I can withdraw from" # Replace with your desired input text
7
8prediction = classifier(text)
9prediction