tangminhanh commited on
Commit
d203cae
·
verified ·
1 Parent(s): 5605ba8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,8 +8,8 @@ token = os.getenv("token")
8
 
9
  # Load the model and tokenizer
10
  model = AutoModelForSequenceClassification.from_pretrained(
11
- "kmcs-casulit/hr_cate", use_auth_token=token)
12
- tokenizer = AutoTokenizer.from_pretrained("kmcs-casulit/department-classification", use_auth_token=token)
13
  pipe = pipeline("text-classification", model=model, tokenizer=tokenizer)
14
 
15
  # Define a request model
 
8
 
9
  # Load the model and tokenizer
10
  model = AutoModelForSequenceClassification.from_pretrained(
11
+ "kmcs-casulit/hr_cate", token=token)
12
+ tokenizer = AutoTokenizer.from_pretrained("kmcs-casulit/department-classification", token=token)
13
  pipe = pipeline("text-classification", model=model, tokenizer=tokenizer)
14
 
15
  # Define a request model