It’s currently looks like this:
func like this:
def predict(img): img = preprocess_image(img) pred = model.predict(img) print(pred) pred = np.squeeze(pred).astype(float) print(pred) return dict(zip(class_names, pred))
