Hugging Face's logo

Spaces:
PVK-Varma
/
Classifier_DistilBERT
Sleeping

App Files Community
PVK-Varma commited on
Commit
a0da6d3
·
verified ·
1 Parent(s): 4ac500d

Update app.py

Browse files

Updated the category size from 3 to 6

Files changed (1) hide show
  1. app.py +1 -4
app.py CHANGED
@@ -5,11 +5,8 @@ import torch
5
 
6
  # Class ID to category name mapping
7
  categories = [
8
- "Transaction",
9
- "Escalation",
10
  "FAQ",
11
- "Account Query",
12
- "Service Request",
13
  "Fallback"
14
  ]
15
 
 
5
 
6
  # Class ID to category name mapping
7
  categories = [
 
 
8
  "FAQ",
9
+ "Escalation",
 
10
  "Fallback"
11
  ]
12