import tensorflow as tf
from load_model import load_model

def predict(text):
    model = load_model()
    prediction = model.predict([text])
    return prediction