File size: 168 Bytes
9a1ddeb |
1 2 3 4 5 6 |
from src.inference.pipeline import load_pipeline
def predict(resume_text, model_path="checkpoints"):
pipe = load_pipeline(model_path)
return pipe(resume_text)
|
9a1ddeb |
1 2 3 4 5 6 |
from src.inference.pipeline import load_pipeline
def predict(resume_text, model_path="checkpoints"):
pipe = load_pipeline(model_path)
return pipe(resume_text)
|