King-8 commited on
Commit
2e88ae2
·
verified ·
1 Parent(s): 49965de

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -6,7 +6,8 @@ import torch.nn.functional as F
6
 
7
  app = FastAPI()
8
 
9
- model_name = "King-8/help-classifier"
 
10
 
11
  tokenizer = AutoTokenizer.from_pretrained(model_name)
12
  model = AutoModelForSequenceClassification.from_pretrained(model_name)
@@ -14,8 +15,8 @@ model = AutoModelForSequenceClassification.from_pretrained(model_name)
14
  labels = [
15
  "learning_help",
16
  "project_help",
17
- "attendance_issue",
18
  "technical_issue",
 
19
  "general_guidance"
20
  ]
21
 
 
6
 
7
  app = FastAPI()
8
 
9
+ # Using v2 model (1000 examples, improved performance)
10
+ model_name = "King-8/help-classifier-v2"
11
 
12
  tokenizer = AutoTokenizer.from_pretrained(model_name)
13
  model = AutoModelForSequenceClassification.from_pretrained(model_name)
 
15
  labels = [
16
  "learning_help",
17
  "project_help",
 
18
  "technical_issue",
19
+ "attendance_issue",
20
  "general_guidance"
21
  ]
22