tensorflow2.0如何根据mnist数据集训练出来的模型，做单个图片预测求教下面怎么写

[sktmeng]

# 下载并安装TensorFlow 2软件包。将TensorFlow导入程序中：
from __future__ import absolute_import, division, print_function, unicode_literals
# Install TensorFlow
import tensorflow as tf
from keras.applications.imagenet_utils import decode_predictions
import matplotlib.pyplot as plt
from PIL import Image

# 保存模型：
keras_model_path = "keras_save"
# 现在，从该文件重新加载模型：
model = tf.keras.models.load_model(keras_model_path)
#单个图片预测
data_dir = r'D:\images\shuzitest\shuzi\IM10.jpg'