Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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",
|
12 |
-
tokenizer = AutoTokenizer.from_pretrained("kmcs-casulit/department-classification",
|
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
|