Pulastya0 commited on
Commit
8560891
·
verified ·
1 Parent(s): f156d8f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -31,7 +31,9 @@ tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, use_fast=True)
31
  model = AutoModelForSequenceClassification.from_pretrained(MODEL_NAME)
32
 
33
  # Departments mapping (example, can adjust for hackathon)
34
- DEPARTMENTS = ["Networking", "Hardware", "Software", "Security", "General IT"]
 
 
35
 
36
  # -------------------------------
37
  # Routing Endpoint
 
31
  model = AutoModelForSequenceClassification.from_pretrained(MODEL_NAME)
32
 
33
  # Departments mapping (example, can adjust for hackathon)
34
+ DEPARTMENTS = ['Account', 'Software', 'Network', 'Security', 'Hardware',
35
+ 'Infrastructure', 'Licensing', 'Communication', 'RemoteWork',
36
+ 'Training', 'Performance']
37
 
38
  # -------------------------------
39
  # Routing Endpoint