Update app.py

app.py

@@ -27,7 +27,7 @@ def predict(input_dict):
     image = image.numpy().reshape(-1, 28, 28, 1)
 
     # Predict
-    prediction = model_cnn.predict(image)
+    prediction = model.predict(image)
     probs = tf.nn.softmax(prediction[0]).numpy()
     return {str(i): float(probs[i]) for i in range(10)}