Update app.py
Browse files
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 = [
|
|
|
|
|
|
|
| 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
|