Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -7,7 +7,7 @@ app = FastAPI()
|
|
7 |
# Load the model and tokenizer
|
8 |
model = AutoModelForSequenceClassification.from_pretrained(
|
9 |
"kmcs-casulit/hr_cate")
|
10 |
-
tokenizer = AutoTokenizer.from_pretrained("kmcs-casulit/
|
11 |
pipe = pipeline("text-classification", model=model, tokenizer=tokenizer)
|
12 |
|
13 |
# Define a request model
|
|
|
7 |
# Load the model and tokenizer
|
8 |
model = AutoModelForSequenceClassification.from_pretrained(
|
9 |
"kmcs-casulit/hr_cate")
|
10 |
+
tokenizer = AutoTokenizer.from_pretrained("kmcs-casulit/department-classification")
|
11 |
pipe = pipeline("text-classification", model=model, tokenizer=tokenizer)
|
12 |
|
13 |
# Define a request model
|