kairaamilanii commited on
Commit
237d67c
1 Parent(s): 80922f5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -4,7 +4,7 @@ import torch
4
 
5
  # Initialize ASR and classifier pipelines
6
  model_asr = "kairaamilanii/whisper-mind14-enUS"
7
- model_class = "kairaamilanii/RoBERTa-minds-14-classifier-intent"
8
 
9
  transcriber = pipeline(
10
  "automatic-speech-recognition",
@@ -25,9 +25,9 @@ intent_classes = {
25
  7: 'cash_deposit',
26
  8: 'direct_debit',
27
  9: 'freeze',
28
- 10: 'latest_transactions',
29
  11: 'joint_account',
30
- 12: 'high_value_payment',
31
  13: 'pay_bill'
32
  }
33
 
 
4
 
5
  # Initialize ASR and classifier pipelines
6
  model_asr = "kairaamilanii/whisper-mind14-enUS"
7
+ model_class = "kairaamilanii/RoBERTa-minds14-en"
8
 
9
  transcriber = pipeline(
10
  "automatic-speech-recognition",
 
25
  7: 'cash_deposit',
26
  8: 'direct_debit',
27
  9: 'freeze',
28
+ 10: 'high_value_payment',
29
  11: 'joint_account',
30
+ 12: 'latest_transactions',
31
  13: 'pay_bill'
32
  }
33