api/inference.py

import joblib

model = joblib.load(
    "models/best_logistic.pkl"
)

vectorizer = joblib.load(
    "models/tfidf_vectorizer.pkl"
)


def predict_claim(text):

    vector = vectorizer.transform(
        [text]
    )

    prediction = model.predict(
        vector
    )

    return str(prediction[0])