Textwizai / utils.py
Erfan11's picture
Update utils.py
0a6e300 verified
raw
history blame
164 Bytes
import tensorflow as tf
from load_model import load_model
def predict(text):
model = load_model()
prediction = model.predict([text])
return prediction