| from model import load_model | |
| import json | |
| import numpy as np | |
| classifier = load_model() | |
| with open("label_list.json","r") as infile: | |
| actions = list(json.load(infile).values()) | |
| def model_predict(input): | |
| """Perform prediction on input (numpy array), return a label (str)""" | |
| res = classifier.predict(np.expand_dims(input,axis=0))[0] | |
| label = actions[np.argmax(res)] | |
| return label |