Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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-
|
| 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: '
|
| 29 |
11: 'joint_account',
|
| 30 |
-
12: '
|
| 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 |
|