src/inference.py

from transformers import pipeline

def infer_resume(model, tokenizer, resume_path):
    with open(resume_path, 'r') as f:
        resume_text = f.read()
    clf = pipeline("text-classification", model=model, tokenizer=tokenizer)
    result = clf(resume_text)
    print("Resume classification:", result)