app.py

from transformers import AutoTokenizer, AutoModelForTokenClassification, pipeline
import torch

# Load the tokenizer and model
tokenizer = AutoTokenizer.from_pretrained("jjzha/jobbert_knowledge_extraction")
model = AutoModelForTokenClassification.from_pretrained("jjzha/jobbert_knowledge_extraction")